IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
e3b879ad3ba0929f720f79f0aa9d66b01f445e0d
openvino/model-optimizer/extensions/ops
History
Vladimir Gavrilov da47cb05be Extend MO for the operation NonMaxSuppression-5 (#2356) 
* Commit.

* Written draft of NonMaxSuppression-5 class.

* Written conversion of the value of the second output  of MO NonMaxSuppression-5 into TF format.

* Fixed type infer for the port 1 of NonMaxSuppression-5.

* Added Reshape to [1] for 0D inputs of NMS-5.

* Small fix.

* Corrected assert for number of inputs.

* Fixed docstrings for transformations TFNonMaxSuppressionNormalize and NonMaxSuppressionNormalize.

* Now the transformation TFNonMaxSuppressionNormalize uses find_and_replace_pattern().

* Moved model-optimizer/extensions/front/onnx/non_max_suppression_normalize.py to model-optimizer/extensions/front/non_max_suppression_normalize.py, to delete duplicate code.

* Deleted commented code.

* Fixed BOM-file.

* Deleted out_ports_count from NMS.

* Fixes in type_infer of NMS-5.

* Small changes.

* Added some comment.

* Small fix.

* Some fixes.
2020-11-06 21:00:55 +03:00
..
__init__.py
Publishing R3
2018-10-16 13:45:03 +03:00
accum_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
accum.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
activation_ops.py
ONNX Loop operation support (#2756)
2020-10-27 23:04:43 +03:00
activation_test.py
Fix Mish and SoftPlus value propagation functions (#2120)
2020-09-11 12:58:14 +03:00
adaptive_avg_pooling.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
argmax_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
argmax.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
assert_op.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
assert_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
aten.py
Implement support for opset3 EmbeddingBag ops (#546)
2020-06-08 18:06:40 +03:00
axpy.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
binarization.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
BlockLSTM.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
bn.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
box_nms.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
bucketize_test.py
Implement Bucketize in MO and MKLDNN for opset3 (#583)
2020-05-28 11:11:07 +03:00
bucketize.py
Implement Bucketize in MO and MKLDNN for opset3 (#583)
2020-05-28 11:11:07 +03:00
Cast.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
constant_fill.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
copyop.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
correlation_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
correlation.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
ctc_greedy_decoder_test.py
Extend MO for operation CTCLoss and partly refactor CTCGreedyDecoder (#588)
2020-08-17 19:19:59 +03:00
ctc_greedy_decoder.py
Implement reshapeable CTCGreedyDecoderPlusSparseToDense transformation and test (#1906)
2020-08-28 14:28:32 +03:00
ctc_loss_test.py
Extend MO for operation CTCLoss and partly refactor CTCGreedyDecoder (#588)
2020-08-17 19:19:59 +03:00
ctc_loss.py
Extend MO for operation CTCLoss and partly refactor CTCGreedyDecoder (#588)
2020-08-17 19:19:59 +03:00
cumsum_test.py
publish master branch snapshot, revision 8d31237e2c3f673cbb0f0ba110fc10f5cce1d2bb
2020-05-22 02:23:12 +03:00
cumsum.py
publish master branch snapshot, revision 8d31237e2c3f673cbb0f0ba110fc10f5cce1d2bb
2020-05-22 02:23:12 +03:00
data_augmentation_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
data_augmentation.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
depth_to_space_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
depth_to_space.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
dequantize_linear.py
Add ONNX DequantizeLinear to MO (#1250)
2020-07-24 18:39:09 +03:00
detectionoutput_onnx.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
DetectionOutput.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
div_value_propagation_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
elementwise_test.py
Add Round-5 operation (#2328)
2020-10-20 18:36:19 +03:00
elementwise.py
Add Round-5 operation (#2328)
2020-10-20 18:36:19 +03:00
embedding_bag_test.py
Implement support for opset3 EmbeddingBag ops (#546)
2020-06-08 18:06:40 +03:00
embedding_bag.py
Implement support for opset3 EmbeddingBag ops (#546)
2020-06-08 18:06:40 +03:00
Enter.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
Exit.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
ExtractImagePatches_test.py
ExtractImagePatches MO implementation and nGraph transformation (#739)
2020-08-19 13:23:41 +03:00
ExtractImagePatches.py
ExtractImagePatches MO implementation and nGraph transformation (#739)
2020-08-19 13:23:41 +03:00
fake_output.py
Move FakeOutput resolving to back phase (#2033)
2020-09-07 10:20:24 +03:00
fakequantize.py
[MO] Fix many spelling and grammar errors (#2543)
2020-10-09 12:16:12 +03:00
gather_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
gather.py
[ MO ] Complete weights layout permutation (#2299)
2020-09-18 14:42:16 +03:00
gathernd_test.py
Extend MO for operation GatherND (#2540)
2020-10-15 07:40:58 +03:00
gathernd.py
Extend MO for operation GatherND (#2540)
2020-10-15 07:40:58 +03:00
GatherTree.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
gelu.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
grn_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
grn.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
GRU.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
GRUCell.py
publish master branch snapshot, revision 59af1853ca21ea08acf17b177da0b239753deb46
2020-05-18 17:21:58 +03:00
hard_sigmoid.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
identity.py
[ MO TF ] IdentityN support (#529)
2020-05-25 10:52:58 +03:00
instance_normalization_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
instance_normalization.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
interp.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
interpolate_test.py
Extend MO for support of Interpolate-4 (#2026)
2020-09-09 16:28:52 +03:00
interpolate.py
Extend MO for support of Interpolate-4 (#2026)
2020-09-09 16:28:52 +03:00
LookupTableInsert_test.py
Implement LookupTableInsert shape inference (#2348)
2020-10-20 09:57:55 +03:00
LookupTableInsert.py
Implement LookupTableInsert shape inference (#2348)
2020-10-20 09:57:55 +03:00
loop.py
ONNX Loop operation support (#2756)
2020-10-27 23:04:43 +03:00
lstm_cell.py
LSTMCell/Sequence v1, reference implementations and decompose transformations for LSTM/GRU/RNN Cells (#2000)
2020-09-04 09:04:36 +03:00
lstm_sequence.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
LSTM.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
MatMul_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
MatMul_value_propagation_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
MatMul.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
merge_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
merge.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
mvn.py
LayerNorm(PyTorch/HuggingFace pattern)->MVN+Mul+Add (#1003)
2020-06-25 09:25:56 +03:00
mxrepeat.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
mxreshape.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
NextIteration.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
non_max_suppression_test.py
NMS-4 op support (#1115)
2020-06-30 14:04:31 +03:00
non_max_suppression.py
Extend MO for the operation NonMaxSuppression-5 (#2356)
2020-11-06 21:00:55 +03:00
non_zero.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
normalize_l2.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
normalize_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
normalize.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
one_hot_test.py
Leaky relu transformation refactor (#2640)
2020-10-14 16:43:29 +03:00
one_hot.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
ONNXResize11.py
Extend MO for support of Interpolate-4 (#2026)
2020-09-09 16:28:52 +03:00
pack.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
parameter.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
pnorm.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
power_file.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
prediction_heatmap.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
prelu.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
priorbox_clustered_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
priorbox_clustered.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
priorbox_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
priorbox.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
priorgridgenerator_onnx.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
proposal_onnx.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
proposal_python_example.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
proposal_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
proposal.py
add preliminary support of Proposal-4 in nGraph (#1448)
2020-08-16 15:49:49 +03:00
psroipooling_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
psroipooling.py
Cleanup IR v7 from the MO (#1008)
2020-06-22 11:52:00 +03:00
quantize_linear.py
[MO] Support ONNX QuantizeLinear (#1451)
2020-08-10 21:10:45 +03:00
quantize_test.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
range.py
Add Range-4 support (#1552)
2020-08-19 07:58:41 +03:00
rank.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
ReduceOps_test.py
Enable ReduceL1 and ReduceL2 operations (#1799)
2020-08-17 16:32:53 +03:00
ReduceOps.py
Enable ReduceL1 and ReduceL2 operations (#1799)
2020-08-17 16:32:53 +03:00
regionyolo_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
regionyolo.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
reorgyolo_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
reorgyolo.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
resample.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
resize_factor_utils.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
resize.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
reverse_sequence.py
publish master branch snapshot, revision 59af1853ca21ea08acf17b177da0b239753deb46
2020-05-18 17:21:58 +03:00
Reverse_test.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
Reverse.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
RNN.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
RNNCell.py
publish master branch snapshot, revision 59af1853ca21ea08acf17b177da0b239753deb46
2020-05-18 17:21:58 +03:00
roialign.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
roifeatureextractor_onnx.py
Add 'aligned' param to ExperimentalDetectronROIFeatureExtractor for CPU plugin and MO
2020-06-03 17:52:40 +03:00
scatter.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
scatternd_test.py
Implement ScatterND operation in MO and transform for SparseToDense (#584)
2020-07-10 12:29:15 +03:00
scatternd.py
Implement ScatterND operation in MO and transform for SparseToDense (#584)
2020-07-10 12:29:15 +03:00
select_test.py
Fix unit tests for select layer. (#638)
2020-06-08 18:39:40 +03:00
select.py
Fix unit tests for select layer. (#638)
2020-06-08 18:39:40 +03:00
shufflechannel.py
publish master branch snapshot, revision 8d31237e2c3f673cbb0f0ba110fc10f5cce1d2bb
2020-05-22 02:23:12 +03:00
simplernms_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
simplernms.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
size.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
slice_like_test.py
[MO MXNET] Fixed spatial reshape on GluonCV models (#587)
2020-06-30 22:32:13 +03:00
slice_like.py
[MO MXNET] Fixed spatial reshape on GluonCV models (#587)
2020-06-30 22:32:13 +03:00
space_to_depth_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
space_to_depth.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
sparse_fill_empty_rows_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
sparse_fill_empty_rows.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
sparse_reshape_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
sparse_reshape.py
[MO] Fix many spelling and grammar errors (#2543)
2020-10-09 12:16:12 +03:00
sparse_segment_mean_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
sparse_segment_mean.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
sparse_segment_sqrtn_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
sparse_segment_sqrtn.py
[MO] Fix many spelling and grammar errors (#2543)
2020-10-09 12:16:12 +03:00
sparse_segment_sum_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
sparse_segment_sum.py
[MO] Fix many spelling and grammar errors (#2543)
2020-10-09 12:16:12 +03:00
spatial_transformer_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
spatial_transformer.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
splice.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
split_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
split.py
[MO] Fix many spelling and grammar errors (#2543)
2020-10-09 12:16:12 +03:00
stop_gradient.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
swapaxis.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
switch_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
switch.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
tensor_iterator.py
ONNX Loop operation support (#2756)
2020-10-27 23:04:43 +03:00
TensorArray.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
TensorArrayGather.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
TensorArrayRead.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
TensorArrayScatter.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
TensorArraySize.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
TensorArrayWrite.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
TensorIterator_ops.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
topk_test.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
topk.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
topkrois_onnx.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
transpose_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
transpose.py
[ MO ] Complete weights layout permutation (#2299)
2020-09-18 14:42:16 +03:00
unique_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
unique.py
[MO] Fix many spelling and grammar errors (#2543)
2020-10-09 12:16:12 +03:00
upsample_test.py
publish master branch snapshot, revision 0110d9c98fd7209589d06344f0d836f61d81f4b3
2020-04-27 21:21:29 +03:00
upsample.py
Improve support ONNX Resize-10 created by PyTorch (#1350)
2020-10-29 15:26:23 +03:00