upd: video dtpe transform from torchvision

openvinotoolkit · Mar 25, 2024 · a76e7f3 · a76e7f3
1 parent dabd0ff
commit a76e7f3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/anomalib/data/base/video.py b/src/anomalib/data/base/video.py
@@ -10,6 +10,7 @@
 import torch
 from pandas import DataFrame
 from torchvision.transforms.v2 import Transform
+from torchvision.transforms.v2.functional._misc import to_dtype_video
 from torchvision.tv_tensors import Mask
 
 from anomalib import TaskType
@@ -153,7 +154,7 @@ def __getitem__(self, index: int) -> dict[str, str | torch.Tensor]:
             msg = "self.indexer must be an instance of ClipsIndexer."
             raise TypeError(msg)
         item = self.indexer.get_item(index)
-        item["image"] = item["image"].to(torch.float32)
+        item["image"] = to_dtype_video(item["image"])
         # include the untransformed image for visualization
         item["original_image"] = item["image"].to(torch.uint8)