MCPcopy Create free account
hub / github.com/VincentHancoder/SSGD / multiclass_nms

Function multiclass_nms

mmdet/core/post_processing/bbox_nms.py:8–95  ·  view source on GitHub ↗

NMS for multi-class bboxes. Args: multi_bboxes (Tensor): shape (n, #class*4) or (n, 4) multi_scores (Tensor): shape (n, #class), where the last column contains scores of the background class, but this will be ignored. score_thr (float): bbox threshold, bboxes

(multi_bboxes,
                   multi_scores,
                   score_thr,
                   nms_cfg,
                   max_num=-1,
                   score_factors=None,
                   return_inds=False)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 11

_get_bboxes_singleMethod · 0.90
_bbox_post_processMethod · 0.90
get_bboxesMethod · 0.90
aug_test_bboxesMethod · 0.90
aug_testMethod · 0.90
aug_testMethod · 0.90
aug_testMethod · 0.90
aug_test_bboxesMethod · 0.90
get_bboxesMethod · 0.90
get_bboxesMethod · 0.90
simple_testMethod · 0.90

Calls 2

catMethod · 0.80
expandMethod · 0.45

Tested by 1

aug_test_bboxesMethod · 0.72