diff --git a/examples/torch/semantic_segmentation/configs/icnet_camvid_int8.json b/examples/torch/semantic_segmentation/configs/icnet_camvid_int8.json index 0b09ca9f565..2c88e050a7c 100644 --- 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"] - } -} \ No newline at end of file + }, + "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 index 093a6a885da..69dcaf70e01 100644 --- 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"] } - ] -} \ No newline at end of file + ], + "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 index 2c21cdee78d..ac72a26cb62 100644 --- 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 index 6cbde3a2622..b4277327ee5 100644 --- 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" } - ] -} \ No newline at end of file + ], + "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 index 9b0acd25d12..243073a24c3 100644 --- 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 index afd93c87715..4dce335be48 100644 --- 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 index 4f78cb2000a..55b86592d7d 100644 --- 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