Add: fix image transform to float

Signed-off-by: Bepitic <[email protected]>
openvinotoolkit · Mar 24, 2024 · dabd0ff · dabd0ff
1 parent a07d47e
commit dabd0ff
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/anomalib/data/base/video.py b/src/anomalib/data/base/video.py
@@ -153,6 +153,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)
         # include the untransformed image for visualization
         item["original_image"] = item["image"].to(torch.uint8)