Training RT-DETR with custom dataset

[viethoang303]

问题确认 Search before asking

• 我已经搜索过问题，但是没有找到解答。I have searched the question and found no related answer.

请提出你的问题 Please ask your question

Hi authors,
I am interested in RT-DETR, and I want to run code to train model but I don't know format of dataset for dataloader. Does it have to have the same format as COCO ?
Thank you

[lyuwenyu]

Your can train rtdetr on your data following below two steps:

1. Prepare your data as coco format, PrepareDetDataSet
2. Modify default dataset config, coco_detection

[viethoang303]

@lyuwenyu Thank you very much, I have just run my code but when I set "export CUDA_VISIBLE_DEVICES=1", the code access to gpu 0 not gpu 1. So how to fix this?

Your can train rtdetr on your data following below two steps:

1. Prepare your data as coco format, PrepareDetDataSet
2. Modify default dataset config, coco_detection