Update phase3pipeline.py

phase3pipeline.py

@@ -171,13 +171,19 @@ class Phase3Pipeline:
 
 
         # Lower threshold for debugging
-        disc_mask = (mean_probs[1] > 0.20).astype(np.uint8)
-        cup_mask = (mean_probs[2] > 0.20).astype(np.uint8)
+        disc_mask = (mean_probs[1] > 0.25).astype(np.uint8)
+        cup_mask = (mean_probs[2] > 0.25).astype(np.uint8)
+
+        print("Raw disc pixels:", int(disc_mask.sum()))
+        print("Raw cup pixels :", int(cup_mask.sum()))
     
 
         # TEMP: disable cleanup during debugging
-        # disc_mask = self._clean_mask(disc_mask)
-        # cup_mask = self._clean_mask(cup_mask)
+        disc_mask = self._clean_mask(disc_mask)
+        cup_mask = self._clean_mask(cup_mask)
+
+        print("Clean disc pixels:", int(disc_mask.sum()))
+        print("Clean cup pixels :", int(cup_mask.sum()))
 
         return (
             disc_mask,