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

pytorch 訓練有复現嗎 #123

Open twmht opened 1 year ago

twmht commented 1 year ago

运行了一下 PyTorch 的训练，觉得收敛速度慢，看结果目前的模型都是从 PP 转过来的，请问 @lyuwenyu 有用 PyTorch 复现出 PP 的结果了吗？

lyuwenyu commented 1 year ago

可以复现 参考下面的log https://github.com/lyuwenyu/RT-DETR/files/13059685/log_rtdetr_r50vd_coco_pytorch.txt

twmht commented 1 year ago

我使用自己的数据集进行训练，但准确度与 YOLO 相差很大

twmht commented 1 year ago

自己训练一个类别，最终 loss 降到约6左右，看起来挺正常，但验证集的 mAP 却比 YOLOX 低很多，你有什么想法吗

twmht commented 1 year ago

对了一下你给的 log，coco 确定没问题，是不是这个方法对于资料量有一定程度的要求? 我的训练资料不到一万张

lyuwenyu commented 1 year ago

对比低很多 有具体的数嘛， 你训练的时候加了预训练参数嘛

我们在很多下游业务数据上已经验证过的 没遇到啥问题啊

twmht commented 1 year ago

mAP 差了 40% 以上，有加预训练参数，也有尝试加上编码器跟解码器的预训练参数，看起来训练的loss 有收敛，但验证很差。请问你们下游业务数据有开放下载的吗？有训练单类别吗？

lyuwenyu commented 1 year ago

确认一下是不是后处理里类别对不上，另外可视化看下box

NeKoooo233 commented 10 months ago

可以复现 参考下面的log https://github.com/lyuwenyu/RT-DETR/files/13059685/log_rtdetr_r50vd_coco_pytorch.txt

这个是用四张卡训练的吗，我算了下，batchsize是16，要是用一张卡来复现，要改成16吗

ainawen commented 6 months ago

可以复现 参考下面的log https://github.com/lyuwenyu/RT-DETR/files/13059685/log_rtdetr_r50vd_coco_pytorch.txt

请问这是加载预训练的resnet50主干的coco成绩吗？如果不加载预训练主干，从零开始训练进行coco数据集上的复现是不是会有精度上的损失？