Add metadata to optimized model (#2618)

* bug fix for legacy openvino models * Add metadata to optimized model * Revert formatting changes --------- Co-authored-by: Ashwin Vaidya <ashwinitinvaidya@gmail.com>
openvinotoolkit · Nov 9, 2023 · 68e42d0 · 68e42d0
1 parent 0d9dff8
commit 68e42d0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/otx/algorithms/anomaly/tasks/openvino.py b/src/otx/algorithms/anomaly/tasks/openvino.py
@@ -361,6 +361,8 @@ def optimize(
         output_model.optimization_type = ModelOptimizationType.POT
         output_model.optimization_methods = [OptimizationMethod.QUANTIZATION]
         output_model.precision = [ModelPrecision.INT8]
+        metadata = self.get_metadata()
+        output_model.set_data("metadata", json.dumps(metadata).encode())
 
         self.task_environment.model = output_model
         self.inference_model = self.get_openvino_model()