RoiSet facilitates object detection models

• created distinct generator class methods for RoiSet
• RoiSet.from_bounding_boxes can create RoiSet with just YXWH coordinates and no segmentation, e.g. as output from object detection models
• can also inspect and quantify overlap between and within RoiSet(s), at both bounding box and segmentation level
• can generate RoiSet from polygon segmentation e.g. in COCO segmentation models

model_server/base/roiset.py

@@ -122,7 +122,8 @@ def filter_df(df: pd.DataFrame, filters: RoiFilter = None) -> pd.DataFrame:
 
 def filter_df_overlap_bbox(df: pd.DataFrame) -> pd.DataFrame:
     """
-    Return subset of DataFrame whose bounding boxes overlap in 3D space
+    Return subset of DataFrame whose bounding boxes overlap in 3D space, with possible duplicate entries where a ROI
+    overlaps with multiple neighbors.
     """
 
     def _compare(r0, r1):
@@ -137,6 +138,8 @@ def filter_df_overlap_bbox(df: pd.DataFrame) -> pd.DataFrame:
         if _compare(df.iloc[pair[0]], df.iloc[pair[1]]):
             first.append(pair[0])
             second.append(pair[1])
+            first.append(pair[1])
+            second.append(pair[0])
 
     sdf = df.iloc[first]
     sdf['bbox_overlaps_with'] = second
@@ -145,7 +148,7 @@ def filter_df_overlap_bbox(df: pd.DataFrame) -> pd.DataFrame:
 # TODO: get overlapping segments
 def filter_df_overlap_seg(df: pd.DataFrame) -> pd.DataFrame:
     """
-    Return subset of DataFrame whose segmentations overlap in 3D space
+    Return subset of DataFrame whose segmentations overlap in 3D space.
     """
 
     dfbb = filter_df_overlap_bbox(df)