最近由于课题研究需要,针对遥感影像需要做旋转目标检测,因此做了yolov5的旋转框改造
代码地址:https://github.com/onehahaha756/yolov5_rotation
1.主要需要修改的地方
- 数据的表示方式
- 代码的dataloader
- 数据增广调整
- 模型的detect模块
- 旋转框的iou、nms
- loss修改
最近由于课题研究需要,针对遥感影像需要做旋转目标检测,因此做了yolov5的旋转框改造
代码地址:https://github.com/onehahaha756/yolov5_rotation