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.31k stars 259 forks source link

Image size for custom dataset #229

Closed BenjaminJonghyun closed 6 months ago

BenjaminJonghyun commented 6 months ago

Hello, thank you for sharing codes.

I'd like to train your model on my custom dataset, which has 640x480(width, height) size. The image size should be fixed to keep aspect ration of the original size.

So, which lines in codes should be modified? In my insight, dataloader.yml L14 and L33 are set to be [480, 640] rtdetr_r50vd.yml L43 and L58 are set to be [480, 640]

please check it.

lyuwenyu commented 6 months ago

Yes.