diff --git a/requirements/anomaly.txt b/requirements/anomaly.txt index 4716c2b33dc..97bbf4db2ba 100644 --- 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 index 23afa946179..81240f31217 100644 --- 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),