yolov5 rotation_yolov5改旋转框进行目标检测记录

文章目录[隐藏]

最近由于课题研究需要,针对遥感影像需要做旋转目标检测,因此做了yolov5的旋转框改造

代码地址:https://github.com/onehahaha756/yolov5_rotation

1.主要需要修改的地方

  • 数据的表示方式
  • 代码的dataloader
  • 数据增广调整
  • 模型的detect模块
  • 旋转框的iou、nms
  • loss修改

版权声明:本文为CSDN博主「one_hahaha」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/weixin_45881512/article/details/119421889

one_hahaha

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐