项目背景
https://blog.csdn.net/x1131230123/article/details/140606459
似乎这个任务是简单的,利用目标检测是否可以完成得好呢?
生成数据集
利用这个代码产生数据集:
为了将标签转换为YOLOv5格式,需要将左上角和右下角的坐标转换为YOLO格式的中心点坐标和宽高。YOLOv5标签格式是这样的:
\text{ {class}} , \text{ {x_center}} , \text{ {y_center}} , \text{ {width}} , \text{ {height}}
其中,(\text{ {x_center}})、(\text{ {y_center}})、(\text{ {width}})和(\text{ {height}})都是相对于图片宽度和高度的归一化值。以下是修改后的代码:
import os
import random