角度计算

21 May, 2023

角度计算

YOLOv5-7.0实例分割训练自己的数据，切分mask图并摆正_jin__9981的博客-CSDN博客

这篇博客介绍了如何使用YOLOv5-7.0进行实例分割训练，并展示了如何切分mask图并摆正。该博客提供了详细的步骤和代码示例。

在 predict.py 的第 170 行添加以下代码：

                    im0_h, im0_w, im0_c = im0.shape
                    for k, seg_list in enumerate(segments):
                        # 将归一化的点转换为坐标点
                        new_seg_list = []
                        for s_point in seg_list:
                            pt1, pt2 = s_point
                            new_pt1 = int(pt1 * im0_w)
                            new_pt2 = int(pt2 * im0_h)
                            new_seg_list.append([new_pt1, new_pt2])
                        rect = cv2.minAreaRect(np.array(new_seg_list))  # 得到最小外接矩形的（中心(x,y), (宽,高), 旋转角度）
                        jiaodu_path =Path(str(save_dir) + "\\jiaodu.txt")
                        with open(jiaodu_path,'a') as f:
                            f.write(p.name + ': ')
                            f.write('{:.2f}'.format(rect[2]))
                            f.write('\n')

https://theailearner.com/tag/cv2-minarearect/#:~:text=OpenCV%20provides%20a%20function%20cv2.minAreaRect%20%28%29%20for%20finding,of%20rotation%29.%20The%20syntax%20is%20given%20below.%201

算法旋转角度解释