uses PIL to calculate image size

Author: lgvaz

icevision/imports.py

@@ -1,5 +1,6 @@
 import sys, os, re, shutil, typing, itertools, operator, math, warnings, json, random
 import functools, io, cv2, mimetypes, torch, torchvision, dataclasses, zipfile, pickle
+import PIL
 from copy import copy, deepcopy
 
 from pdb import set_trace

icevision/utils/imageio.py

@@ -8,10 +8,15 @@
 ]
 
 from icevision.imports import *
-import PIL, imagesize
+from PIL import ExifTags
 
 ImgSize = namedtuple("ImgSize", "width,height")
 
+# get exif tag
+for _EXIF_ORIENTATION_TAG in ExifTags.TAGS.keys():
+    if PIL.ExifTags.TAGS[_EXIF_ORIENTATION_TAG] == "Orientation":
+        break
+
 
 def open_img(fn, gray=False):
     color = "L" if gray else "RGB"
@@ -27,14 +32,26 @@ def get_image_size(filepath: Union[str, Path]) -> Tuple[int, int]:
     Returns image (width, height)
     """
     logger.warning("get_image_size is deprecated, use get_img_size instead")
-    return imagesize.get(filepath)
+    image_size = get_img_size(filepath=filepath)
+    return image_size.width, image_size.height
 
 
 def get_img_size(filepath: Union[str, Path]) -> ImgSize:
     """
     Returns image (width, height)
     """
-    return ImgSize(*imagesize.get(filepath))
+    with PIL.Image.open(filepath) as image:
+        image_size = image.size
+
+    try:
+        exif = image._getexif()
+        if exif is not None and exif[_EXIF_ORIENTATION_TAG] in [6, 8]:
+            image_size = image_size[::-1]
+    except (AttributeError, KeyError, IndexError):
+        # cases: image don't have getexif
+        pass
+
+    return ImgSize(*image_size)
 
 
 def show_img(img, ax=None, show: bool = False, **kwargs):

setup.cfg

@@ -35,7 +35,6 @@ install_requires =
     pycocotools>=2.0.2<3
     requests >=2.24.0<3
     loguru >=0.5.3
-    imagesize >= 1.2.0
     pillow > 8.0.0
     importlib-metadata>=1;python_version<"3.8"
 

tests/utils/test_imageio.py

@@ -6,7 +6,7 @@
     "fn,expected",
     [
         ("voc/JPEGImages/2007_000063.jpg", (375, 500, 3)),
-        ("images2/flies.jpeg", (2592, 3888, 3)),
+        ("images2/flies.jpeg", (3888, 2592, 3)),
     ],
 )
 def test_open_img(samples_source, fn, expected):
@@ -18,7 +18,7 @@ def test_open_img(samples_source, fn, expected):
     "fn,expected",
     [
         ("voc/JPEGImages/2007_000063.jpg", (500, 375)),
-        ("images2/flies.jpeg", (3888, 2592)),
+        ("images2/flies.jpeg", (2592, 3888)),
     ],
 )
 def test_get_image_size(samples_source, fn, expected):