↓ 2 callersMethodget_losses(
self,
imgs,
x_shifts,
y_shifts,
expanded_strides,
labels,
unicorn/models/yolo_head_det.py:331
↓ 2 callersMethodget_losses(
self,
imgs,
x_shifts,
y_shifts,
expanded_strides,
labels,
unicorn/models/unicorn_head.py:484
↓ 2 callersFunctionget_mosaic_coordinate(mosaic_image, mosaic_index, xc, yc, w, h, input_h, input_w)
unicorn/data/datasets/mosaicdetection.py:17
↓ 2 callersMethodget_summary(accs, names, metrics=('mota', 'num_switches', 'idp', 'idr', 'idf1', 'precision', 'recall'))
unicorn/evaluators/evaluation.py:90
↓ 2 callersMethodmask_heads_forward_with_coords mask_feats: (1, 8, H//8, W//8), mask_feat_stride=8, mask_head_params: (N, 169), instance_locations: (N, 2), instance_fpn_levels: (N,
unicorn/models/condinst/dynamic_mask_head.py:172
↓ 2 callersMethodmatch(self, bboxes, labels, track_feats, frame_id, asso_tau=-1, return_index=False)
unicorn/tracker/quasi_dense_embed_tracker.py:137
↓ 2 callersFunctionrandom_perspective(
img,
targets=(),
degrees=10,
translate=0.1,
scale=0.1,
shear=10,
perspective=0.0
unicorn/data/data_augment.py:52
↓ 2 callersFunctionwindow_partition Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
unicorn/models/backbone/swin_transformer.py:39