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: https://github.com/huggingface/transformers/pull/33244