Add new notebook with egsis and point annotator (#31)

Fix behavior of `egsis.labeling.get_superpixel_label` by ignoring the unlabeled pixels in the histogram to found the most ocurrent label.
ryukinix · Oct 19, 2023 · 99bbd8f · 99bbd8f
1 parent 6d2c2f7
commit 99bbd8f
Show file tree

Hide file tree

Showing 3 changed files with 289 additions and 55 deletions.
diff --git a/egsis/labeling.py b/egsis/labeling.py
@@ -31,9 +31,10 @@ def get_superpixel_label(
     superpixels: numpy.ndarray,
     superpixel: int,
 ):
-    """Get the most ocurrent label in superpixel"""
-    histogram = numpy.bincount(label_matrix[superpixels == superpixel].flatten())
-    most_occurent_label_at_superpixel = numpy.argmax(histogram)
+    """Get the most ocurrent label in superpixel, ignoring unlabeled"""
+    y = label_matrix[(label_matrix != 0) & (superpixels == superpixel)]
+    histogram = numpy.bincount(y.flatten())
+    most_occurent_label_at_superpixel = numpy.argmax(histogram) if histogram.size > 0 else 0
 
     return most_occurent_label_at_superpixel
 

diff --git a/notebooks/annotator_egsis.ipynb b/notebooks/annotator_egsis.ipynb
diff --git a/notebooks/model_egsis.ipynb b/notebooks/model_egsis.ipynb