From 5b23f1db063ab8a89a1ea96af7a479dded0e3c3e Mon Sep 17 00:00:00 2001 From: Ashwin Vaidya Date: Thu, 22 Sep 2022 10:17:22 +0200 Subject: [PATCH] Check for successful openvino conversion --- anomalib/deploy/optimize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/anomalib/deploy/optimize.py b/anomalib/deploy/optimize.py index a91fcc08ca..93b433da5b 100644 --- a/anomalib/deploy/optimize.py +++ b/anomalib/deploy/optimize.py @@ -68,7 +68,7 @@ def export_convert( if export_mode == "openvino": export_path = os.path.join(str(export_path), "openvino") optimize_command = "mo --input_model " + str(onnx_path) + " --output_dir " + str(export_path) - os.system(optimize_command) + assert os.system(optimize_command) == 0, "OpenVINO conversion failed" with open(Path(export_path) / "meta_data.json", "w", encoding="utf-8") as metadata_file: meta_data = get_model_metadata(model) # Convert metadata from torch