↓ 7 callersFunctionnmsNon-maximum suppression.
Arguments:
boxes: a float numpy array of shape [n, 5],
where each row is (xmin, ymin, xmax, ymax
models/mtcnn/mtcnn_pytorch/src/box_utils.py:5
↓ 6 callersMethod__init__(self, *, image_size, patch_size, num_classes, dim, depth, heads, mlp_dim, channels = 3, attn_dropout = 0., ff
vit_pytorch/vit_pytorch.py:81