Update inference.py

inference.py

@@ -62,7 +62,16 @@ def prediction_mask(prediction_result, map_name):
     wid, hight = prediction_result.shape[0], prediction_result.shape[1]
     mask = np.zeros([wid, hight])
     mask = cv2.fillPoly(mask, pts=[contour], color=(1,1,1)).astype(int)
-    masked_img = cv2.bitwise_and(prediction_result, mask)
+    
+    try:
+        masked_img = cv2.bitwise_and(prediction_result, mask)
+    except cv2.error as e:
+        print("Error with cv2.bitwise_and:")
+        print("prediction_result shape:", prediction_result.shape)
+        print("mask shape:", mask.shape)
+        print("prediction_result dtype:", prediction_result.dtype)
+        print("mask dtype:", mask.dtype)
+        raise e
 
     return masked_img