↓ 6 callersFunctionpermute_to_N_HWA_K Transpose/reshape a tensor from (N, (A x K), H, W) to (N, (HxWxA), K)
models/retinanet/retinanet.py:32
↓ 3 callersMethod__init__(self, in_channels, conv_channels, num_convs, pred_channels, pred_prior=None)
models/querydet/det_head.py:92
↓ 1 callersMethoddet_loss(self, gt_classes, gt_anchors_targets, pred_logits, pred_deltas, all_anchors)
models/querydet/detector.py:274
↓ 1 callersMethoddet_loss(self, gt_classes, gt_anchors_deltas, pred_logits, pred_deltas, alphas, gammas, cls_weights, reg_weights)
models/retinanet/retinanet.py:273
↓ 1 callersFunctioneval_res
:param gt0: np.array[ng, 5], ground truth results [x, y, w, h, ignore]
:param dt0: np.array[nd, 5], detection results [x, y, w, h, score]
visdrone_eval/viseval/calc_accuracy.py:5