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

Method get_label_assignment

mmdet/models/dense_heads/lad_head.py:16–119  ·  view source on GitHub ↗

Get label assignment (from teacher). Args: cls_scores (list[Tensor]): Box scores for each scale level. Has shape (N, num_anchors * num_classes, H, W) bbox_preds (list[Tensor]): Box energies / deltas for each scale level with shape (N,

(self,
                             cls_scores,
                             bbox_preds,
                             iou_preds,
                             gt_bboxes,
                             gt_labels,
                             img_metas,
                             gt_bboxes_ignore=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

test_lad_head_lossFunction · 0.95
forward_trainMethod · 0.80

Calls 5

multi_applyFunction · 0.90
catMethod · 0.80
levels_to_imagesFunction · 0.70
get_anchorsMethod · 0.45
get_targetsMethod · 0.45

Tested by 1

test_lad_head_lossFunction · 0.76