RT-DETR icon indicating copy to clipboard operation
RT-DETR copied to clipboard
Published 4 months ago • verified publisher icon lyuwenyu

旋转框预测该如何修改

Open ThreeStones1029 opened this issue 2 years ago • 3 comments

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

ThreeStones1029 avatar Jan 03 '24 10:01 ThreeStones1029

暂时没有这方面的计划

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

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

lyuwenyu avatar Jan 04 '24 06:01 lyuwenyu

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

ThreeStones1029 avatar Jan 09 '24 14:01 ThreeStones1029

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

Li-Yidong avatar Sep 11 '24 03:09 Li-Yidong