旋转目标检测rotation-yolov5笔记

https://github.com/root12321/Rotation-Detect-yolov5_poly

这个也算一个:

实现可选多边形信息输出。通过扩展输出维度,额外输出多边形顶点坐标信息,相对box框左上角,归一化至box宽高的信息实现多边形信息的输出。
在整个多边形计算中,从数据结构,数据增强,模型结构,输出结构,坐标后处理,损失计算,NMS过滤都做了相应的调整。
目前只支持输出指定边数(edges参数)的多边形,如果需要识别不同边数多边形,可将数据集最后一个点进行重复处理。


GitHub - HRan2004/Yolo-ArbV2

本项目主要基于yolov5算法的旋转目标检测,主要借鉴项目https://github.com/hukaixuan19970627/YOLOv5_DOTA_OBB

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

AI视觉网奇

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

暂无评论

发表评论

相关推荐

深度学习之目标检测YOLOv5

一.简介 YOLOV4出现之后不久,YOLOv5横空出世。YOLOv5在YOLOv4算法的基础上做了进一步的改进,检测性能得到进一步的提升。虽然YOLOv5算法并没有与YOLOv4算法进行性能比较与分析&#xff0