Skip to content

Commit 00a84df

Browse files
ashwinvaidya17Ashwin Vaidya
andauthored
Bug fix for albumentations (#2467)
* bug fix for legacy openvino models * Address albumentation issue --------- Co-authored-by: Ashwin Vaidya <[email protected]>
1 parent 2fd7c3e commit 00a84df

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/otx/algorithms/anomaly/tasks/openvino.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,8 @@ def _populate_metadata_legacy(self, model: ModelEntity) -> Dict[str, Any]:
242242
to_tensor=True,
243243
)
244244
metadata = {
245-
"transform": transform.to_dict(),
245+
# TODO: Replace with transform.to_dict() when OTX supports albumentations 1.3.0
246+
"transform": {"transform": transform._to_dict()},
246247
"image_threshold": image_threshold,
247248
"pixel_threshold": pixel_threshold,
248249
"min": min_value,

0 commit comments

Comments
 (0)