add min_bbox_visibility to loader

Author: sokovninn

luxonis_ml/data/loaders/luxonis_loader.py

@@ -48,6 +48,7 @@ def __init__(
         exclude_empty_annotations: bool = False,
         color_space: Literal["RGB", "BGR", "GRAY"] = "RGB",
         seed: int | None = None,
+        min_bbox_visibility: float = 0.0,
         bbox_area_threshold: float = 0.0004,
         *,
         keep_categorical_as_strings: bool = False,
@@ -96,6 +97,8 @@ def __init__(
             to C{"RGB"}.
         @type seed: Optional[int]
         @param seed: The random seed to use for the augmentations.
+        @type min_bbox_visibility: float
+        @param min_bbox_visibility: Minimum fraction of the original bounding box that must remain visible after augmentation.
         @type bbox_area_threshold: float
         @param bbox_area_threshold: Minimum area threshold for bounding boxes to be considered valid. In the range [0, 1].
             Default is 0.0004, which corresponds to a small area threshold to remove invalid bboxes and respective keypoints.
@@ -222,6 +225,7 @@ def __init__(
             width,
             keep_aspect_ratio,
             seed,
+            min_bbox_visibility,
             bbox_area_threshold,
         )
 
@@ -439,6 +443,7 @@ def _init_augmentations(
         width: int | None,
         keep_aspect_ratio: bool,
         seed: int | None = None,
+        min_bbox_visibility: float = 0.0,
         bbox_area_threshold: float = 0.0004,
     ) -> AugmentationEngine | None:
         if isinstance(augmentation_config, PathType):
@@ -481,6 +486,7 @@ def _init_augmentations(
             keep_aspect_ratio=keep_aspect_ratio,
             is_validation_pipeline="train" not in self.view,
             seed=seed,
+            min_bbox_visibility=min_bbox_visibility,
             bbox_area_threshold=bbox_area_threshold,
         )