Make anomaly task compatible with older albumentations versions (#2363)

* fix transforms export in metadata * wrap transform dict * add todo for updating to_dict call
openvinotoolkit · Jul 17, 2023 · 895bd36 · 895bd36
1 parent c1c0463
commit 895bd36
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements/anomaly.txt b/requirements/anomaly.txt
@@ -1,6 +1,5 @@
 # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
 # Anomaly Requirements.
-albumentations>=1.3.0
 torchvision<0.15.1
 torchtext<0.15.1
 anomalib==0.5.1

diff --git a/src/otx/algorithms/anomaly/tasks/inference.py b/src/otx/algorithms/anomaly/tasks/inference.py
@@ -425,7 +425,8 @@ def _get_metadata_dict(self) -> Dict[str, Any]:
             else:
                 transform = self.trainer.datamodule.train_dataloader().dataset.transform
         metadata = {
-            "transform": transform.to_dict(),
+            # TODO: Replace with transform.to_dict() when OTX supports albumentations 1.3.0
+            "transform": {"transform": transform._to_dict()},
             "image_threshold": image_threshold,
             "pixel_threshold": pixel_threshold,
             "image_shape": list(self.config.model.input_size),