onnx

openvinotoolkit · Oct 12, 2023 · 07cec83 · 07cec83
1 parent 45d63a2
commit 07cec83
Show file tree

Hide file tree

Showing 7 changed files with 21 additions and 20 deletions.
diff --git a/examples/torch/semantic_segmentation/configs/icnet_camvid_int8.json b/examples/torch/semantic_segmentation/configs/icnet_camvid_int8.json
@@ -42,5 +42,6 @@
     "compression": {
             "algorithm": "quantization",
             "ignored_scopes": ["ICNet/interpolate_0"]
-    }
-}
+    },
+    "export_via_onnx": true
+}
diff --git a/examples/torch/semantic_segmentation/configs/icnet_camvid_magnitude_sparsity_int8.json b/examples/torch/semantic_segmentation/configs/icnet_camvid_magnitude_sparsity_int8.json
@@ -68,5 +68,6 @@
             "algorithm": "quantization",
             "ignored_scopes": ["ICNet/interpolate_0"]
     }
-    ]
-}
+    ],
+    "export_via_onnx": true
+}
diff --git a/examples/torch/semantic_segmentation/configs/unet_camvid_int8.json b/examples/torch/semantic_segmentation/configs/unet_camvid_int8.json
@@ -36,5 +36,6 @@
     },
     "compression": {
             "algorithm": "quantization"
-    }
+    },
+    "export_via_onnx": true
 }
diff --git a/examples/torch/semantic_segmentation/configs/unet_camvid_magnitude_sparsity_int8.json b/examples/torch/semantic_segmentation/configs/unet_camvid_magnitude_sparsity_int8.json
@@ -62,5 +62,6 @@
         {
             "algorithm": "quantization"
     }
-    ]
-}
+    ],
+    "export_via_onnx": true
+}
diff --git a/examples/torch/semantic_segmentation/configs/unet_mapillary_int8.json b/examples/torch/semantic_segmentation/configs/unet_mapillary_int8.json
@@ -13,13 +13,13 @@
     },
 
     "input_info": {
-      "sample_size": [1, 3, 512, 1024]
+        "sample_size": [1, 3, 512, 1024]
     },
 
     "weighing": [0.0000, 0.0554, 0.3358, 0.0843, 0.7865, 0.7753, 1.1966, 5.1031, 2.4255,
         0.0679, 0.8589, 0.0389, 2.8977, 9.4937, 0.2531, 1.8852, 2.1179, 2.1978,
         5.9516, 6.4394],
-    "multiprocessing_distributed" : true,    
+    "multiprocessing_distributed" : true,
     "optimizer": {
         "type": "Adam",
         "optimizer_params": {
@@ -32,9 +32,10 @@
     },
 
     "model_params": {
-         "input_size_hw": [512, 1024]
+        "input_size_hw": [512, 1024]
     },
     "compression": {
-            "algorithm": "quantization"
-    }
+        "algorithm": "quantization"
+    },
+    "export_via_onnx": true
 }
diff --git a/examples/torch/semantic_segmentation/configs/unet_mapillary_magnitude_sparsity_int8.json b/examples/torch/semantic_segmentation/configs/unet_mapillary_magnitude_sparsity_int8.json
@@ -32,7 +32,7 @@
     },
 
     "model_params": {
-         "input_size_hw": [512, 1024]
+        "input_size_hw": [512, 1024]
     },
     "compression": [
         {
@@ -59,6 +59,7 @@
         },
         {
             "algorithm": "quantization"
-    }
-    ]
+        }
+    ],
+    "export_via_onnx": true
 }
diff --git a/tests/torch/data/ac_configs/ssd512_vgg_voc_int8.yml b/tests/torch/data/ac_configs/ssd512_vgg_voc_int8.yml
@@ -29,8 +29,3 @@ models:
             integral: 11point
             ignore_difficult: True
             presenter: print_scalar
-
-          - type: normalization
-            mean: 123.675, 116.28, 103.53
-            std: 58.4795, 57.1429, 57.4713
-          - type: bgr_to_rgb
-Original file line number
+Diff line change
@@ Expand Up / @@ -62,5 +62,6 @@ @@
             {
                 "algorithm": "quantization"
         }
-        ]
-    }
+        ],
+        "export_via_onnx": true
+    }