Dataset class for SoccerNet

ASTRA model class

Class to compute the ASTRA loss (classification + displacement)

Class to implement Learning rate warmup

Dataset class for SoccerNet (inference / testing)

Standard prediction head

Class to compute the loglikelihood loss (displacement loss uncertaint-aware)

Mixup module class

Uncertainty-aware prediction head