diff --git a/supervision/__init__.py b/supervision/__init__.py index 3e5c418c..2e227b57 100644 --- a/supervision/__init__.py +++ b/supervision/__init__.py @@ -79,6 +79,7 @@ from supervision.detection.utils.iou_and_nms import ( box_non_max_suppression, mask_iou_batch, mask_non_max_suppression, + mask_non_max_merge, oriented_box_iou_batch, ) from supervision.detection.utils.masks import ( @@ -222,6 +223,7 @@ __all__ = [ "list_files_with_extensions", "mask_iou_batch", "mask_non_max_suppression", + "mask_non_max_merge", "mask_to_polygons", "mask_to_rle", "mask_to_xyxy",