Update test.py

zhangmozhe · web-flow · commit 53a30232c90e · 2021-07-12T20:01:29.000+08:00
diff --git a/Global/test.py b/Global/test.py
@@ -14,7 +14,7 @@
 import torchvision.transforms as transforms
 import torchvision.transforms as transforms
 import numpy as np
-
+import cv2
 
 def data_transforms(img, method=Image.BILINEAR, scale=False):
 
