准备工作:
a、先个人使用labelimg或者其他标注软件标注一部分数据集
b、训练一个有一定识别能力的yolov5模型
c、先使用模型推理一遍,看效果如何,是否能框上很多目标
d、调整置信度--conf-thres的大小,例如0.1,0.2,0.25,
0.45这类的常用数值,觉得检测效果不错了就可以开始下面的工作了
1、找到yolov5的推理文件detect.py,打开保存txt开关,如图:
这个开关是保存标签的,会将输入的图片经过推理,并将图片和标签一同保存在runs/detect/exp/下面,如图片:
2、在detect.py的163-177行代码,如下图,把图片中的部分代码注解掉,这样就不会在图片上或者结果的框,而是直接复制过去,并保存标签。(或者你能到生成的标签,推理后的图片就不要了,直接和输入的源图片一起作为数据集,这样也是可以的。)