Merge pull request #65 from dvschultz/master

cover use case of .jpeg files
richzhang · Mar 1, 2021 · cbbf945 · cbbf945
2 parents 05f4ec2 + 90949a2
commit cbbf945
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lpips/__init__.py b/lpips/__init__.py
@@ -105,7 +105,7 @@ def load_image(path):
         import rawpy
         with rawpy.imread(path) as raw:
             img = raw.postprocess()
-    elif(path[-3:]=='bmp' or path[-3:]=='jpg' or path[-3:]=='png'):
+    elif(path[-3:]=='bmp' or path[-3:]=='jpg' or path[-3:]=='png' or path[-4:]=='jpeg'):
         import cv2
         return cv2.imread(path)[:,:,::-1]
     else: