update mask_handler

sorghum_pipeline/data/mask_handler.py

@@ -15,7 +15,19 @@ class MaskHandler:
         """Apply mask to image."""
         if mask is None:
             return image
+        
+        # Ensure mask is uint8
+        mask = mask.astype(np.uint8)
+        
+        # Resize mask to match image if needed
         if mask.shape[:2] != image.shape[:2]:
             mask = cv2.resize(mask, (image.shape[1], image.shape[0]), interpolation=cv2.INTER_NEAREST)
+        
+        # Create binary mask
         binary = (mask > 0).astype(np.uint8) * 255
+        
+        # Ensure binary mask is single channel and matches image dimensions
+        if binary.shape[:2] != image.shape[:2]:
+            binary = cv2.resize(binary, (image.shape[1], image.shape[0]), interpolation=cv2.INTER_NEAREST)
+        
         return cv2.bitwise_and(image, image, mask=binary)