🐞 Bugfix: Set reverse_input_channels to True in OpenVINO models (#2848)

* Set reverse_input_channels to True
openvinotoolkit · Jan 29, 2024 · af77ed0 · af77ed0
1 parent 87df806
commit af77ed0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/otx/algorithms/anomaly/tasks/inference.py b/src/otx/algorithms/anomaly/tasks/inference.py
@@ -357,7 +357,7 @@ def _add_metadata_to_ir(self, model_file: str, export_type: ExportType) -> None:
         if "min" in metadata and "max" in metadata:
             extra_model_data[("model_info", "normalization_scale")] = metadata["max"] - metadata["min"]
 
-        extra_model_data[("model_info", "reverse_input_channels")] = False
+        extra_model_data[("model_info", "reverse_input_channels")] = True  # convert BGR to RGB
         extra_model_data[("model_info", "model_type")] = "AnomalyDetection"
 
         labels = []

diff --git a/src/otx/algorithms/anomaly/tasks/openvino.py b/src/otx/algorithms/anomaly/tasks/openvino.py
@@ -453,7 +453,7 @@ def _metadata_in_ir_format(self) -> Dict[Tuple[str, str], Union[str, int, float,
         if "min" in metadata and "max" in metadata:
             extra_model_data[("model_info", "normalization_scale")] = metadata["max"] - metadata["min"]
 
-        extra_model_data[("model_info", "reverse_input_channels")] = False
+        extra_model_data[("model_info", "reverse_input_channels")] = True  # convert BGR to RGB
         extra_model_data[("model_info", "model_type")] = "AnomalyDetection"
         extra_model_data[("model_info", "labels")] = "Normal Anomaly"
         return extra_model_data