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.61k stars 303 forks source link

joint training on multiple datasets #377

Open goldentimecoolk opened 3 months ago

goldentimecoolk commented 3 months ago

Hi @lyuwenyu , thanks for the amazing work. I want to jointly train a rtdetr model on multiple datasets. Generally, I found the solution as dataset = torch.utils.data.ConcatDataset([dataset1, dataset2]) with dataset1 = CocoDetection() and dataset2 = CocoDetection(). But in the repo, the code is highly encapsulated. I have no idea to achieve the function. Can you give me some help? Thanks.