From 31bc1271ae6f13b7e281b9959ac24a5e8f2ed522 Mon Sep 17 00:00:00 2001
From: Richard Zhang <rzhang88@gmail.com>
Date: Fri, 1 Oct 2021 09:16:03 -0700
Subject: [PATCH] Update __init__.py

---
 lpips/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lpips/__init__.py b/lpips/__init__.py
index 9dc89877..89e5bba2 100755
--- 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