预测

WeberKang · Dec 22, 2020 · fd086f0 · fd086f0
1 parent 63df256
commit fd086f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/predict.py b/predict.py
@@ -16,7 +16,7 @@
 
         image = cv2.imread(imagePath)
         output = image.copy()
-        image = cv2.resize(image, (224, 224))
+        image = cv2.resize(image, (224, 224))#根据所选网络对其修改
 
         # scale图像数据
         image = image.astype("float32") / 255.0