Fixed PIL size bug in ImageRecordComponent (#889) (#894)

airctic · Jul 14, 2021 · 944b47c · 944b47c
1 parent 0088ad2
commit 944b47c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/icevision/core/record_components.py b/icevision/core/record_components.py
@@ -115,7 +115,7 @@ def set_img(self, img: Union[PIL.Image.Image, np.ndarray]):
         assert isinstance(img, (PIL.Image.Image, np.ndarray))
         self.img = img
         if isinstance(img, PIL.Image.Image):
-            height, width = img.shape
+            width, height = img.size
         elif isinstance(img, np.ndarray):
             # else:
             height, width, _ = self.img.shape
@@ -136,7 +136,7 @@ def _repr(self) -> List[str]:
                     )
                 return [f"Img: {width}x{height}x{channels} <np.ndarray> Image"]
             elif isinstance(self.img, PIL.Image.Image):
-                height, width = self.img.shape
+                width, height = self.img.size
                 return [f"Img: {width}x{height} <PIL.Image; mode='{self.img.mode}'>"]
         else:
             return [f"Img: {self.img}"]