add getting nms_threshold / iou_threshold from RetinaNet (#3075)
authorPavel Esir <pavel.esir@intel.com>
Thu, 12 Nov 2020 12:04:07 +0000 (15:04 +0300)
committerGitHub <noreply@github.com>
Thu, 12 Nov 2020 12:04:07 +0000 (15:04 +0300)
commit8c89d8d7331ed4ebbc0ad69578fd6e31f0b668d1
tree46e86d9edb4e2f1b162503ded5a5d72798a4c167
parent0f4525affcc33ac26848313f1fa20652c69e0c07
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
model-optimizer/extensions/front/tf/RetinaNetFilteredDetectionsReplacement.py
model-optimizer/extensions/front/tf/non_max_suppression_ext.py
model-optimizer/extensions/front/tf/retinanet.json
model-optimizer/extensions/ops/non_max_suppression.py