From: Anton Chetverikov <Anton.Chetverikov@intel.com>
Date: Thu, 27 Aug 2020 10:10:44 +0000 (+0300)
Subject: Update priorbox extenders (#1925)
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=675e1c821ec10aa144e127c28a70ba38e07a372a;p=platform%2Fupstream%2Fdldt.git

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
--- 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
--- 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)