Can we have rdetr pretained mode?

[niiish32x]

I can't find rtdetr model like rtdetr_l.pt. There is only rtdetr model with suffix .pdparams.

[Laughing-q]

@niiish32x just use the following script and the weight will automatically download.

from ultralytics.vit import RTDETR
model = RTDETR("rtdetr-x.pt")

[niiish32x]

Thanks a lot!
By the way I hope i can use ultralytics framework to train my own rtdetr mode one day.

[glenn-jocher]

@niiish32x you're welcome! We're glad to hear about your interest in training your own RT-DETR model. While we currently don't offer custom training for RT-DETR directly within the Ultralytics framework, we're continuously working on improving and expanding our capabilities. Keep an eye on our updates, and thank you for being part of the YOLO community! 🚀