文章目录[隐藏]
最近由于课题研究需要,针对遥感影像需要做旋转目标检测,因此做了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
暂无评论