Most of the state is available, now just load the image crops.
MOTImage.arbitrary_to_ds()
dgs.models.dataset.MOT.MOTImage
dgs.models.dataset.MOT.MOTImage.configure_torch_module