search

lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
2.62k stars 302 forks source link

training pytorch RT-DETR on single-class object detection #73

Open chau25102001 opened 1 year ago

chau25102001 commented 1 year ago

Hello and thanks for your work. I have a question regarding training RT-DETR on a custom dataset with only one object class (e.g. Face). Does this repo work with single-class object detection? If so what should I change in the code?

Thanks.

lyuwenyu commented 1 year ago

Yes, It can handle single-class case