旋转目标检测rotation-yolov5笔记

目录

钟表项目：

预测角度：

ncnn推理代码

修改的地方

从头开始训练及测试流程（代码使用说明)

损失函数部分：

---

---

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

这个也算一个：

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

修改代码：

def img2label_paths(img_paths):
    # Define label paths as a function of image paths
    sa, sb = os.sep + 'imgs' + os.sep, os.sep + 'labels_poly' + os.sep  # /images/, /labels/ substrings
    # sa, sb = os.sep + 'img_rotate' + os.sep, os.sep + 'labels_rotate' + os.sep  # /imag