IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update priorbox extenders (#1925)

Browse Source
This commit is contained in:
Anton Chetverikov 2020-08-27 13:10:44 +03:00 committed by GitHub
parent a4d90a0aba
commit 675e1c821e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model-optimizer/mo/utils/ir_reader/extenders/priorbox_clustered_extender.py
View File

@ -27,3 +27,5 @@ class PriorBoxClustered_extender(Extender):
op['V10_infer'] = True
PriorBox_extender.attr_restore(op, 'width', value=1.0)
PriorBox_extender.attr_restore(op, 'height', value=1.0)
PriorBox_extender.attr_restore(op, 'variance')

2
model-optimizer/mo/utils/ir_reader/extenders/priorbox_extender.py
View File

@ -26,7 +26,7 @@ class PriorBox_extender(Extender):
def extend(op: Node):
op['V10_infer'] = True
attrs = ['min_size', 'max_size', 'aspect_ratio']
attrs = ['min_size', 'max_size', 'aspect_ratio', 'variance', 'fixed_ratio', 'fixed_size', 'density']
for attr in attrs:
PriorBox_extender.attr_restore(op, attr)