From 3731d01d7be68dac816d77ff08c884fc7d8ded20 Mon Sep 17 00:00:00 2001 From: Bepitic Date: Tue, 9 Apr 2024 16:13:52 +0200 Subject: [PATCH] Fix test-val dsize.empty Signed-off-by: Bepitic --- src/anomalib/utils/visualization/image.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/anomalib/utils/visualization/image.py b/src/anomalib/utils/visualization/image.py index ddcfd416fb..e2c57c35bc 100644 --- a/src/anomalib/utils/visualization/image.py +++ b/src/anomalib/utils/visualization/image.py @@ -145,6 +145,7 @@ def _visualize_batch(self, batch: dict) -> Iterator[GeneratorResult]: elif "video_path" in batch: height, width = batch["image"].shape[-2:] image = batch["original_image"][i].squeeze().cpu().numpy() + image = image.transpose(1, 2, 0) image = cv2.resize(image, dsize=(width, height), interpolation=cv2.INTER_AREA) else: msg = "Batch must have either 'image_path' or 'video_path' defined."