natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
openvino/model-optimizer/extensions/ops
History
Pavel Esir 8c89d8d733
add getting nms_threshold / iou_threshold from RetinaNet (#3075) 
* added getting nms_threshold/iou_threshold from original TF RetinaNet model

* iou_threshold definition added

* fixed getting iou_threshold for TF NMS V2, some minor corrections

* added box_encoding to NMS extractors
 		2020-11-12 15:04:07 +03:00
..
BlockLSTM.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
Cast.py MO clean from IR v7 and other legacy code (#1521) 2020-07-29 17:43:12 +03:00
DetectionOutput.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +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.py ExtractImagePatches MO implementation and nGraph transformation (#739) 2020-08-19 13:23:41 +03:00
ExtractImagePatches_test.py ExtractImagePatches MO implementation and nGraph transformation (#739) 2020-08-19 13:23:41 +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
GatherTree.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
LSTM.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
LookupTableInsert.py Implement LookupTableInsert shape inference (#2348) 2020-10-20 09:57:55 +03:00
LookupTableInsert_test.py Implement LookupTableInsert shape inference (#2348) 2020-10-20 09:57:55 +03:00
MatMul.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +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
NextIteration.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
ONNXResize11.py retinanet onnx model has Upsample error (#3052) 2020-11-11 17:31:00 +03:00
ONNXResize11_test.py retinanet onnx model has Upsample error (#3052) 2020-11-11 17:31:00 +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
ReduceOps.py Enable ReduceL1 and ReduceL2 operations (#1799) 2020-08-17 16:32:53 +03:00
ReduceOps_test.py Enable ReduceL1 and ReduceL2 operations (#1799) 2020-08-17 16:32:53 +03:00
Reverse.py Publishing 2020.2 content 2020-04-13 21:17:23 +03:00
Reverse_test.py Publishing 2020.2 content 2020-04-13 21:17:23 +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
__init__.py Publishing R3 2018-10-16 13:45:03 +03:00
accum.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
accum_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +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.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
argmax_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +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
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.py Implement Bucketize in MO and MKLDNN for opset3 (#583) 2020-05-28 11:11:07 +03:00
bucketize_test.py Implement Bucketize in MO and MKLDNN for opset3 (#583) 2020-05-28 11:11:07 +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.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
ctc_greedy_decoder.py Implement reshapeable CTCGreedyDecoderPlusSparseToDense transformation and test (#1906) 2020-08-28 14:28:32 +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_loss.py Extend MO for operation CTCLoss and partly refactor CTCGreedyDecoder (#588) 2020-08-17 19:19:59 +03:00
ctc_loss_test.py Extend MO for operation CTCLoss and partly refactor CTCGreedyDecoder (#588) 2020-08-17 19:19:59 +03:00
cumsum.py publish master branch snapshot, revision 8d31237e2c3f673cbb0f0ba110fc10f5cce1d2bb 2020-05-22 02:23:12 +03:00
cumsum_test.py publish master branch snapshot, revision 8d31237e2c3f673cbb0f0ba110fc10f5cce1d2bb 2020-05-22 02:23:12 +03:00
data_augmentation.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
data_augmentation_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
depth_to_space.py MO clean from IR v7 and other legacy code (#1521) 2020-07-29 17:43:12 +03:00
depth_to_space_test.py publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b 2020-04-15 19:01:57 +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
div_value_propagation_test.py publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b 2020-04-15 19:01:57 +03:00
elementwise.py Add Round-5 operation (#2328) 2020-10-20 18:36:19 +03:00
elementwise_test.py Add Round-5 operation (#2328) 2020-10-20 18:36:19 +03:00
embedding_bag.py Implement support for opset3 EmbeddingBag ops (#546) 2020-06-08 18:06:40 +03:00
embedding_bag_test.py Implement support for opset3 EmbeddingBag ops (#546) 2020-06-08 18:06:40 +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.py [ MO ] Complete weights layout permutation (#2299) 2020-09-18 14:42:16 +03:00
gather_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
gathernd.py Extend MO for operation GatherND (#2540) 2020-10-15 07:40:58 +03:00
gathernd_test.py Extend MO for operation GatherND (#2540) 2020-10-15 07:40:58 +03:00
gelu.py Publishing 2020.2 content 2020-04-13 21:17:23 +03:00
grn.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
grn_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +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.py publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b 2020-04-15 19:01:57 +03:00
instance_normalization_test.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.py Extend MO for support of Interpolate-4 (#2026) 2020-09-09 16:28:52 +03:00
interpolate_test.py Extend MO for support of Interpolate-4 (#2026) 2020-09-09 16:28:52 +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
merge.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
merge_test.py publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b 2020-04-15 19:01:57 +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
non_max_suppression.py add getting nms_threshold / iou_threshold from RetinaNet (#3075) 2020-11-12 15:04:07 +03:00
non_max_suppression_test.py NMS-4 op support (#1115) 2020-06-30 14:04:31 +03:00
non_zero.py publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b 2020-04-15 19:01:57 +03:00
normalize.py MO clean from IR v7 and other legacy code (#1521) 2020-07-29 17:43:12 +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
one_hot.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
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.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
priorbox_clustered.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
priorbox_clustered_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
priorbox_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
priorgridgenerator_onnx.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
proposal.py add preliminary support of Proposal-4 in nGraph (#1448) 2020-08-16 15:49:49 +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
psroipooling.py Cleanup IR v7 from the MO (#1008) 2020-06-22 11:52:00 +03:00
psroipooling_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +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
regionyolo.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
regionyolo_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
reorgyolo_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
resample.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
resize_factor_utils.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
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.py Implement ScatterND operation in MO and transform for SparseToDense (#584) 2020-07-10 12:29:15 +03:00
scatternd_test.py Implement ScatterND operation in MO and transform for SparseToDense (#584) 2020-07-10 12:29:15 +03:00
select.py Fix unit tests for select layer. (#638) 2020-06-08 18:39:40 +03:00
select_test.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.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
simplernms_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
size.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
slice_like.py [MO MXNET] Fixed spatial reshape on GluonCV models (#587) 2020-06-30 22:32:13 +03:00
slice_like_test.py [MO MXNET] Fixed spatial reshape on GluonCV models (#587) 2020-06-30 22:32:13 +03:00
space_to_depth.py MO clean from IR v7 and other legacy code (#1521) 2020-07-29 17:43:12 +03:00
space_to_depth_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_fill_empty_rows_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_reshape_test.py publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b 2020-04-15 19:01:57 +03:00
sparse_segment_mean.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
sparse_segment_mean_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_sqrtn_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
sparse_segment_sum_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
spatial_transformer_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
splice.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
split.py [MO] Fix many spelling and grammar errors (#2543) 2020-10-09 12:16:12 +03:00
split_test.py publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b 2020-04-15 19:01:57 +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.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
tensor_iterator.py ONNX Loop operation support (#2756) 2020-10-27 23:04:43 +03:00
topk.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
topk_test.py MO clean from IR v7 and other legacy code (#1521) 2020-07-29 17:43:12 +03:00
topkrois_onnx.py publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687 2020-05-13 21:12:22 +03:00
transpose.py [ MO ] Complete weights layout permutation (#2299) 2020-09-18 14:42:16 +03:00
transpose_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
unique_test.py publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b 2020-04-15 19:01:57 +03:00
upsample.py Improve support ONNX Resize-10 created by PyTorch (#1350) 2020-10-29 15:26:23 +03:00
upsample_test.py publish master branch snapshot, revision 0110d9c98fd7209589d06344f0d836f61d81f4b3 2020-04-27 21:21:29 +03:00