RT-DETRv2 is now available in 🤗 Transformers

[NielsRogge]

Hi,

RT-DETRv2 is now available in 🤗 Transformers (besides RT-DETR which was already available): .

All checkpoints are on the hub: https://huggingface.co/models?other=rt_detr_v2.

It allows easy usage in a few lines of code, easy fine-tuning on a custom dataset, easy optimization using ONNX, TensorRT etc. 🥳

Inference

For inference, I refer to the example code snippet in the docs.

Fine-tuning on custom data

For fine-tuning on a custom dataset, I refer to this demo notebook - it is exacty the same for RT-DETRv2 as all object detection models in the Transformers library have the same API.

Happy fine-tuning!