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

activation_ext_test.py

Add asinh acosh atanh extractors (#1600)

2020-08-04 14:56:15 +03:00

activation_ext.py

ONNX Loop operation support (#2756)

2020-10-27 23:04:43 +03:00

affine_ext_test.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

affine_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

argmax_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

aten_ext.py

Implement support for opset3 EmbeddingBag ops (#546)

2020-06-08 18:06:40 +03:00

AttributedSliceToSlice_test.py

Reshape able slice (#1241)

2020-08-10 12:19:08 +03:00

AttributedSliceToSlice.py

Reshape able slice (#1241)

2020-08-10 12:19:08 +03:00

cast_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

clip_ext.py

Add assert for Clip-11 (#795)

2020-07-06 18:10:14 +03:00

const_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

constant_fill_ext.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

constant_of_shape_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

constant_of_shape_to_broadcast.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

conv_ext_test.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

conv_ext.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

crop_ext_test.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

crop_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

cumsum_ext.py

publish master branch snapshot, revision 8d31237e2c3f673cbb0f0ba110fc10f5cce1d2bb

2020-05-22 02:23:12 +03:00

deformable_conv_ext.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

depth_to_space_ext.py

Add ONNX SpaceToDepth and DepthToSpace extractor (#1122)

2020-07-23 13:05:42 +03:00

dequantize_linear_ext.py

Add ONNX DequantizeLinear to MO (#1250)

2020-07-24 18:39:09 +03:00

dequantize_linear_resolver_test.py

[ MO ] KSO ON by default (#1730)

2020-08-20 12:06:45 +03:00

dequantize_linear_resolver.py

Add ONNX DequantizeLinear to MO (#1250)

2020-07-24 18:39:09 +03:00

detection_output_test.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

detection_output.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

detectionoutput_ext.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

dropout_ext.py

[ MO TF ] IdentityN support (#529)

2020-05-25 10:52:58 +03:00

elementwise_ext.py

ONNX Loop operation support (#2756)

2020-10-27 23:04:43 +03:00

expand_ext.py

publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687

2020-05-13 21:12:22 +03:00

flatten_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

flattenONNX_to_reshape.py

Add LogSoftmax-5 to MO and ngraph (#2409)

2020-10-20 13:40:06 +03:00

gather_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

gathernd_ext.py

Extend MO for operation GatherND (#2540)

2020-10-15 07:40:58 +03:00

gemm_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

group_norm_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

gru_ext_test.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

gru_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

hard_sigmoid_ext.py

Fix const node non-deterministic names (part 1) (#996)

2020-06-26 13:41:49 +03:00

image_scaler_ext_test.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

image_scaler_ext.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

instance_normalization_ext_test.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

instance_normalization_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

logsoftmaxONNX_to_logsoftmax.py

Add LogSoftmax-5 to MO and ngraph (#2409)

2020-10-20 13:40:06 +03:00

loop_ext.py

ONNX Loop operation support (#2756)

2020-10-27 23:04:43 +03:00

LoopNormalize.py

ONNX Loop operation support (#2756)

2020-10-27 23:04:43 +03:00

lp_normalization_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

lrn_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

lstm_ext_test.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

lstm_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

mask_rcnn_conversion.py

Fixes for Mask-RCNN conversion (#654)

2020-05-28 14:31:42 +03:00

mask_rcnn.json

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

matmul_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

mean_variance_normalization_ext.py

Fixed extractor for MVN from ONNX (#1653)

2020-08-06 13:53:16 +03:00

non_max_suppression_ext.py

publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687

2020-05-13 21:12:22 +03:00

non_zero_ext.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

normalize_ext_test.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

normalize_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

normalize_l2_normalize.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

one_hot_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

one_hot_normalize.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

pad_converter_test.py

Add support for ONNX Pad-11 (#744)

2020-06-04 14:48:31 +03:00

pad_converter.py

Add support for ONNX Pad-11 (#744)

2020-06-04 14:48:31 +03:00

pad_ext_test.py

Cleanup IR v7 from the MO (#1008)

2020-06-22 11:52:00 +03:00

pad_ext.py

Add support for ONNX Pad-11 (#744)

2020-06-04 14:48:31 +03:00

parameter_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

person_detection_crossroad_conversion.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

person_detection_crossroad.json

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

pooling_ext.py

publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687

2020-05-13 21:12:22 +03:00

priorbox_clustered_ext_test.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

priorbox_clustered_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

priorbox_ext_test.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

priorbox_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

priorgridgenerator_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

proposal_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

quantize_dequantize_linear_test.py

[ MO ] KSO ON by default (#1730)

2020-08-20 12:06:45 +03:00

quantize_dequantize_linear.py

Cleanup IR v7 from the MO (#1008)

2020-06-22 11:52:00 +03:00

quantize_ext.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

quantize_linear_ext.py

[MO] Support ONNX QuantizeLinear (#1451)

2020-08-10 21:10:45 +03:00

quantize_linear_resolver_test.py

[ MO ] KSO ON by default (#1730)

2020-08-20 12:06:45 +03:00

quantize_linear_resolver.py

[MO] Support ONNX QuantizeLinear (#1451)

2020-08-10 21:10:45 +03:00

range_ext.py

Add Range-4 support (#1552)

2020-08-19 07:58:41 +03:00

reduce_ext.py

Enable ReduceL1 and ReduceL2 operations (#1799)

2020-08-17 16:32:53 +03:00

remove_filtering_boxes_by_size.py

Cleanup IR v7 from the MO (#1008)

2020-06-22 11:52:00 +03:00

resize_ext.py

Extend MO for support of Interpolate-4 (#2026)

2020-09-09 16:28:52 +03:00

resize_to_interpolate.py

Cleanup IR v7 from the MO (#1008)

2020-06-22 11:52:00 +03:00

reverse_sequence_ext.py

[ MO ONNX ] Resize-11 clear error message (#620)

2020-05-27 08:09:15 +03:00

rnn_ext_test.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

rnn_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

roialign_ext.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

roifeatureextractor_ext.py

Add 'aligned' param to ExperimentalDetectronROIFeatureExtractor for CPU plugin and MO

2020-06-03 17:52:40 +03:00

scatter_ext.py

ONNX ScatterND MO support (#1813)

2020-08-20 13:57:22 +03:00

shape_ext.py

publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687

2020-05-13 21:12:22 +03:00

slice_ext.py

Reshape able slice (#1241)

2020-08-10 12:19:08 +03:00

softmax_ext.py

Add LogSoftmax-5 to MO and ngraph (#2409)

2020-10-20 13:40:06 +03:00

softmaxONNX_to_softmax.py

publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687

2020-05-13 21:12:22 +03:00

softplus_ext.py

Imironov/cvs 31297 add yolov4 support (#594)

2020-06-02 19:20:29 +03:00

space_to_depth_ext.py

Add ONNX SpaceToDepth and DepthToSpace extractor (#1122)

2020-07-23 13:05:42 +03:00

split_ext.py

Publishing 2020.2 content

2020-04-13 21:17:23 +03:00

squeeze_ext_test.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

squeeze_ext.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

top_k_ext.py

[ MO ONNX ] TopK-1/10/11 proper extracting (#600)

2020-05-26 21:53:24 +03:00

topkrois_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

transpose_ext_test.py

publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b

2020-04-15 19:01:57 +03:00

transpose_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

unsqueeze_ext_test.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

unsqueeze_ext.py

Publishing 2020.1 content

2020-02-11 22:48:49 +03:00

upsample_ext_test.py

[MO] Fix many spelling and grammar errors (#2543)

2020-10-09 12:16:12 +03:00

upsample_ext.py

[MO] Fix many spelling and grammar errors (#2543)

2020-10-09 12:16:12 +03:00

where_ext.py

publish master branch snapshot, revision 0110d9c98fd7209589d06344f0d836f61d81f4b3

2020-04-27 21:21:29 +03:00