torchsig.models.spectrogram_models.detr.modulesΒΆ

Functions

create_detr

Function used to build a DETR network

Classes

Chunker

ConvDownSampler

DETRModel

HungarianMatcher

This class computes an assignment between the targets and the predictions of the network For efficiency reasons, the targets don't include the no_object.

MLP

Very simple multi-layer perceptron (also called FFN) from DETR repo

SetCriterion

This class computes the loss for DETR. The process happens in two steps: 1) we compute hungarian assignment between ground truth boxes and the outputs of the model 2) we supervise each pair of matched ground-truth / prediction (supervise class and box).

XCiT