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

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

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

1.主要需要修改的地方

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

你可能感兴趣的:(人工智能,计算机视觉,算法,深度学习)