tfm.vision.augment.filter_boxes_by_ioa

Filter boxes by intersection over area (IOA).

tfm.vision.augment.filter_boxes_by_ioa(
    bboxes: tf.Tensor, crop_box: tf.Tensor, min_box_overlap: float
) -> tf.Tensor

The boxes with IOA less than min_box_overlap will be replaced by (0, 0, 0, 0) so they can be filtered out later.

bboxes a float tensor of shape [N, 4] representing normalized bounding box coordinates.
crop_box a float tensor of shape [1, 1, 4] representing the normalized crop box.
min_box_overlap minimum overlap of the box with the crop box to keep the box.

a tensor of shape [N, 4] with filtered box coordinates replaced by 0.