edit May15 5:09PM

inference.py

@@ -134,10 +134,16 @@ def preprocess(image, mask):
     if mask.shape[2]:
         mask = cv2.cvtColor(mask, cv2.COLOR_BGR2GRAY)
 
+    
+
     image = image.astype(np.float32) / 255.0
     mask = mask.astype(np.float32) / 255.0
     mask = (mask > 0.5).astype(np.float32)
-    mask = cv2.resize(mask, (image.shape[0], image.shape[1]))
+    # mask = cv2.resize(mask, (image.shape[0], image.shape[1]))
+
+    image = cv2.resize(image, (512, 512))
+    mask = cv2.resize(mask, (512, 512))
+
 
     mask = np.expand_dims(mask, axis=-1)