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

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

暂无评论

发表评论

相关推荐

AP AR mAP ROC AUC(目标检测)

禁止转载!在做目标检测任务的时候,通常会制定规则来评估性能,就如下图所示,后面会慢慢道来其中的原理。 混淆矩阵中 TP、TN、FP、FN
在目标检测中,通常以IoU阈值作为

RCNN详解

RCNN详解
RCNN即region proposals(候选区域) CNN,是将CNN引入目标检测领域的开山之作(2014年),大大提高了目标检测的效果&