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: