Pavel Esir
4302e2c120
renamed logits -> bbox_deltas
updated ngraph unittests for Proposal
removed validate_and_infer_types Proposal-4
removed validate_and_infer_types Proposal-4
changed validate_and_infer_types in parent class of Proposal
removed get_output_size
successfully inferred Proposal on SSH and Faster-RCNN
added unittests for Proposal-4
added unittests for Proposal-4
added unittests for Proposal-4
returned back default namespace for Proposal
reduced number of outputs in v0::Proposal
correct conversion of Proposal-4 -> propodal_ie with 2 outputs
removed creator for proposal v0
removed converter for proposal v0
added Proposal-4 to MO
removed `for_deformable` attribute
added Proposal-4 to MO and nGraph Python API
removed typo in Proposal-4 specification
style corrections
style corrections and removed some redundant code
rename proposal Python api test
removed 'attrs' context from visitor
returned back AttrVisitor to check if passes OpenVINO ONNX pipeline
Should pass OpenVINO ONNX pipeline (returned back AttrVisitor just to check)
python api for Proposal-4 works ok
(style correction) python api for Proposal-4 works ok
parametrized proposal_ie some other corrections
removed 'attrs.' context from nGraph Python API tests for Proposal
minor corrections in replacer proposal->proposal_ie
corrected Python API OpenVINO-ONNX tests should pass
Improved workaround for AttributeVisitor for Proposal
Add additional check of im_info tensor shape to Proposal node in MKLDNNPlugin
😠 removed 4 extra spaces from test_dyn_attributes.py to match The Style
added new nGraph RTTI declarations, removed throwing exception in transformation
added new nGraph RTTI declarations, removed throwing exception in transformation, corrected exception in MKLDNNplugin
corrected im_info size checking in Proposal node of MKLDNNPlugin
81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
"""
|
|
Copyright (C) 2018-2020 Intel Corporation
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
"""
|
|
|
|
from mo.front.common.partial_infer.utils import int64_array
|
|
from mo.front.extractor import attr_getter
|
|
from mo.graph.graph import Node, Graph
|
|
from mo.ops.op import Op
|
|
|
|
|
|
class ProposalOp(Op):
|
|
op = 'Proposal'
|
|
|
|
def __init__(self, graph: Graph, attrs: dict):
|
|
mandatory_props = {
|
|
'type': __class__.op,
|
|
'op': __class__.op,
|
|
'version': 'opset4',
|
|
'post_nms_topn': 300, # default in caffe-shared
|
|
'infer': ProposalOp.proposal_infer,
|
|
'in_ports_count': 3,
|
|
'out_ports_count': 2,
|
|
'normalize': 0,
|
|
}
|
|
super().__init__(graph, mandatory_props, attrs)
|
|
|
|
def supported_attrs(self):
|
|
return [
|
|
'feat_stride',
|
|
'base_size',
|
|
'min_size',
|
|
'ratio',
|
|
'scale',
|
|
'pre_nms_topn',
|
|
'post_nms_topn',
|
|
'nms_thresh',
|
|
]
|
|
|
|
def backend_attrs(self):
|
|
return [
|
|
'feat_stride',
|
|
'base_size',
|
|
'min_size',
|
|
('ratio', lambda node: attr_getter(node, 'ratio')),
|
|
('scale', lambda node: attr_getter(node, 'scale')),
|
|
'pre_nms_topn',
|
|
'post_nms_topn',
|
|
'nms_thresh',
|
|
'framework',
|
|
'box_coordinate_scale',
|
|
'box_size_scale',
|
|
'normalize',
|
|
'clip_after_nms',
|
|
'clip_before_nms',
|
|
]
|
|
|
|
@staticmethod
|
|
def proposal_infer(node: Node):
|
|
input_shape = node.in_node(0).shape
|
|
out_shape = int64_array([input_shape[0] * node.post_nms_topn, 5])
|
|
# rois blob: holds R regions of interest, each is a 5 - tuple
|
|
# (n, x1, y1, x2, y2) specifying an image batch index n and a
|
|
# rectangle(x1, y1, x2, y2)
|
|
node.out_port(0).data.set_shape(out_shape)
|
|
|
|
# the second optional output contains box probabilities
|
|
if len(node.out_ports()) == 2 and not node.out_port(1).disconnected():
|
|
node.out_port(1).data.set_shape(int64_array([input_shape[0] * node.post_nms_topn]))
|