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

旋转框预测该如何修改 #175

Open ThreeStones1029 opened 8 months ago

ThreeStones1029 commented 8 months ago

你好，如果我想修改为旋转框的检测该如何修改，请问是否有这方面的打算？

lyuwenyu commented 8 months ago

暂时没有这方面的计划

适配的话我个人觉得应该改几个方面

  1. dataset & dataloader
  2. box head改成支持旋转框的形式
  3. cross attn改成在多边形里采样点
  4. 后处理
  5. 评价指标
ThreeStones1029 commented 8 months ago

好的，谢谢您的回复，我会尝试去修改。

Li-Yidong commented 2 weeks ago

@ThreeStones1029 您好，我最近也在尝试修改旋转框的检测，请问您成功了吗？