Update classes.py

danforthcenter · Jun 4, 2024 · 615c0f8 · 615c0f8
1 parent b370e68
commit 615c0f8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/plantcv/annotate/classes.py b/plantcv/annotate/classes.py
@@ -216,8 +216,10 @@ def correct_mask(self, bin_img):
                 print("filtering un-annotated object from the mask")
                 labeled_mask[np.where(labeled_mask == i)] = 0
 
-        #completed_mask = np.where(labeled_mask > 0, 255, 0)
-        completed_mask = np.copy(labeled_mask)
+        completed_mask_bin = np.where(labeled_mask > 0, 255, 0)
+        labeled_mask_all, _ = create_labels(mask=completed_mask_bin)
+        completed_mask = np.copy(labeled_mask_all)
+        pix_vals, counts = np.unique(completed_mask, return_counts=True)
 
         object_count = 0
         # points in class used for recovering and labeling