Update __init__.py

richzhang · Oct 1, 2021 · 31bc127 · 31bc127
1 parent c82f1ed
commit 31bc127
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lpips/__init__.py b/lpips/__init__.py
@@ -73,6 +73,7 @@ def load_image(path):
  import cv2
  return cv2.imread(path)[:,:,::-1]
  else:
+ import matplotlib.pyplot as plt 
  img = (255*plt.imread(path)[:,:,:3]).astype('uint8')
 
  return img