(boxlist, min_size)
return boxlist[keep]
Only keep boxes with both sides >= min_size

    Arguments:
        boxlist (Boxlist)
        min_size (int)
Only keep boxes with both sides >= min_size
Only keep boxes with both sides > = min_size
def remove_small_boxes(boxlist, min_size):
    """
    Only keep boxes with both sides >= min_size

    Arguments:
        boxlist (Boxlist)
        min_size (int)
    """
    # TODO maybe add an API for querying the ws / hs
    xywh_boxes = boxlist.convert("xywh").bbox
    _, _, ws, hs = xywh_boxes.unbind(dim=1)
    keep = (
        (ws >= min_size) & (hs >= min_size)
    ).nonzero().squeeze(1)
    return boxlist[keep]
def remove_small_boxes ( boxlist , min_size ) : # TODO maybe add an API for querying the ws / hs xywh_boxes = boxlist . convert ( "xywh" ) . bbox _ , _ , ws , hs = xywh_boxes . unbind ( dim = 1 ) keep = ( ( ws >= min_size ) & ( hs >= min_size ) ) . nonzero ( ) . squeeze ( 1 ) return boxlist [ keep ]