RT-DETR is now available in Hugging Face Transformers

[NielsRogge]

Hi there,

Since a few months, the RT-DETR is officially supported by the Hugging Face Transformers library 🤗

It enables easy inference as well as fine-tuning.

Resources

• RT-DETR docs: https://huggingface.co/docs/transformers/main/en/model_doc/rt_detr
• Demo notebooks: https://github.com/NielsRogge/Transformers-Tutorials/tree/master/RT-DETR
• Official object detection example scripts: https://github.com/huggingface/transformers/tree/main/examples/pytorch/object-detection. These leverage either the 🤗 Trainer API or 🤗 Accelerate, depending on what you prefer for fine-tuning.

Happy fine-tuning!

Related issues

#394 => with the scripts above, multi-GPU fine-tuning works out-of-the-box.
#108

[umarbutler]

Does this support RTDETRv2?

[NielsRogge]

@umarbutler that's WIP here: huggingface/transformers#33244