Detr: Demonstration DETR training code

[matyasbohacek]

Hi,

I wanted to ask whether the training code for the simplified demonstration DETR (Demo Colab here) is available publicly? In this example, only a model pretrained on COCO is loaded and there is no reference whatsoever to the specifics of the training or the code itself.

I have been trying to implement this on my own using the torch's CocoDetection dataset, but I have been running into many issues for multiple days, still not making much progress.

I would very much appreciate any help on this! Otherwise thanks for a great set of tools to experiment with the architecture.

[lhuasheng]

You can try this link.

https://www.kaggle.com/tanulsingh077/end-to-end-object-detection-with-transformers-detr

[matyasbohacek]

Thanks, this definitely helped! I managed to write my own script for it.