Update utils.py

bubbliiiing · web-flow · commit 45ec8d1b655e · 2021-10-16T13:24:38.000+08:00
diff --git a/utils/utils.py b/utils/utils.py
@@ -6,7 +6,7 @@
 #   代码仅仅支持RGB图像的预测，所有其它类型的图像都会转化成RGB
 #---------------------------------------------------------#
 def cvtColor(image):
-    if len(np.shape(image)) == 3 and np.shape(image)[-2] == 3:
+    if len(np.shape(image)) == 3 and np.shape(image)[2] == 3:
         return image 
     else:
         image = image.convert('RGB')
@@ -59,4 +59,4 @@ def get_lr(optimizer):
 
 def preprocess_input(image):
     image /= 255.0
-    return image
+    return image
