lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

旋转框预测该如何修改

ThreeStones1029 opened this issue · comments

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

暂时没有这方面的计划

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

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

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