torchsig.models.spectrogram_models.detr.utilsΒΆ

Functions

accuracy

Computes the precision@k for the specified values of k

box_cxcywh_to_xyxy

box_iou

drop_classifier

find_output_features

format_preds

format_targets

generalized_box_iou

Generalized IoU from https://giou.stanford.edu/ The boxes should be in [x0, y0, x1, y1] format Returns a [N, M] pairwise matrix, where N = len(boxes1) and M = len(boxes2)

get_world_size

is_dist_avail_and_initialized

xcit_name_to_timm_name