Update priorbox extenders (#1925)

diff --git a/model-optimizer/mo/utils/ir_reader/extenders/priorbox_clustered_extender.py b/model-optimizer/mo/utils/ir_reader/extenders/priorbox_clustered_extender.py
index 600f586..c259f98 100644 (file)
--- a/model-optimizer/mo/utils/ir_reader/extenders/priorbox_clustered_extender.py
+++ b/model-optimizer/mo/utils/ir_reader/extenders/priorbox_clustered_extender.py
@@ -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')

diff --git a/model-optimizer/mo/utils/ir_reader/extenders/priorbox_extender.py b/model-optimizer/mo/utils/ir_reader/extenders/priorbox_extender.py
index 4b535cc..2bb34d2 100644 (file)
--- a/model-optimizer/mo/utils/ir_reader/extenders/priorbox_extender.py
+++ b/model-optimizer/mo/utils/ir_reader/extenders/priorbox_extender.py
@@ -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)