Changed configs for SSD models (openvinotoolkit#654)

evgeniya-egupova · Apr 16, 2021 · e2e3cda · e2e3cda
1 parent 6466330
commit e2e3cda
Show file tree

Hide file tree

Showing 10 changed files with 13 additions and 11 deletions.
diff --git a/tests/data/ac_configs/ssd300_mobilenet_voc.yml b/tests/data/ac_configs/ssd300_mobilenet_voc.yml
@@ -21,6 +21,6 @@ models:
           - type: resize_prediction_boxes
         metrics:
           - type: map
-            integral: max
+            integral: 11point
             ignore_difficult: True
             presenter: print_scalar
diff --git a/tests/data/ac_configs/ssd300_mobilenet_voc_magnitude_sparsity_int8.yml b/tests/data/ac_configs/ssd300_mobilenet_voc_magnitude_sparsity_int8.yml
@@ -21,6 +21,6 @@ models:
           - type: resize_prediction_boxes
         metrics:
           - type: map
-            integral: max
+            integral: 11point
             ignore_difficult: True
             presenter: print_scalar
diff --git a/tests/data/ac_configs/ssd300_vgg_voc.yml b/tests/data/ac_configs/ssd300_vgg_voc.yml
@@ -21,6 +21,6 @@ models:
           - type: resize_prediction_boxes
         metrics:
           - type: map
-            integral: max
+            integral: 11point
             ignore_difficult: True
             presenter: print_scalar
diff --git a/tests/data/ac_configs/ssd300_vgg_voc_int8.yml b/tests/data/ac_configs/ssd300_vgg_voc_int8.yml
@@ -21,6 +21,6 @@ models:
           - type: resize_prediction_boxes
         metrics:
           - type: map
-            integral: max
+            integral: 11point
             ignore_difficult: True
             presenter: print_scalar
diff --git a/tests/data/ac_configs/ssd300_vgg_voc_magnitude_sparsity_int8.yml b/tests/data/ac_configs/ssd300_vgg_voc_magnitude_sparsity_int8.yml
@@ -21,6 +21,6 @@ models:
           - type: resize_prediction_boxes
         metrics:
           - type: map
-            integral: max
+            integral: 11point
             ignore_difficult: True
             presenter: print_scalar
diff --git a/tests/data/ac_configs/ssd300_vgg_voc_pruning_geometric_median.yml b/tests/data/ac_configs/ssd300_vgg_voc_pruning_geometric_median.yml
@@ -21,6 +21,6 @@ models:
           - type: resize_prediction_boxes
         metrics:
           - type: map
-            integral: max
+            integral: 11point
             ignore_difficult: True
             presenter: print_scalar
diff --git a/tests/data/ac_configs/ssd512_vgg_voc.yml b/tests/data/ac_configs/ssd512_vgg_voc.yml
@@ -22,6 +22,6 @@ models:
           - type: resize_prediction_boxes
         metrics:
           - type: map
-            integral: max
+            integral: 11point
             ignore_difficult: True
             presenter: print_scalar
diff --git a/tests/data/ac_configs/ssd512_vgg_voc_int8.yml b/tests/data/ac_configs/ssd512_vgg_voc_int8.yml
@@ -22,6 +22,6 @@ models:
           - type: resize_prediction_boxes
         metrics:
           - type: map
-            integral: max
+            integral: 11point
             ignore_difficult: True
             presenter: print_scalar
diff --git a/tests/data/ac_configs/ssd512_vgg_voc_magnitude_sparsity_int8.yml b/tests/data/ac_configs/ssd512_vgg_voc_magnitude_sparsity_int8.yml
@@ -22,6 +22,6 @@ models:
           - type: resize_prediction_boxes
         metrics:
           - type: map
-            integral: max
+            integral: 11point
             ignore_difficult: True
             presenter: print_scalar
diff --git a/tests/sota_checkpoints_eval.json b/tests/sota_checkpoints_eval.json
@@ -296,7 +296,8 @@
                 "metric_type": "Mean AP",
                 "resume": "ssd300_vgg_voc.pth",
                 "batch": 120,
-                "model_description": "SSD300-VGG-BN"
+                "model_description": "SSD300-VGG-BN",
+                "diff_target_max": 0.3
             },
             "ssd300_vgg_voc_int8": {
                 "config": "examples/object_detection/configs/ssd300_vgg_voc_int8.json",
@@ -338,7 +339,8 @@
                 "metric_type": "Mean AP",
                 "resume": "ssd512_vgg_voc.pth",
                 "batch": 32,
-                "model_description": "SSD512-VGG-BN"
+                "model_description": "SSD512-VGG-BN",
+                "diff_target_max": 0.4
             },
             "ssd512_vgg_voc_int8": {
                 "config": "examples/object_detection/configs/ssd512_vgg_voc_int8.json",