利用yolov5的预测结果,生成txt标注文件(与labelImg一样),实现自动标注

1.下载yolov5代码,或者直接用已有的代码。

2.pycharm终端运行:

#yolov5 detect.py help data
python detect.py -h 


利用yolov5的预测结果,生成txt标注文件(与labelImg一样),实现自动标注_第1张图片

3.使用以下参数,可以将yolov5预测的结果,转为txt格式。注意哈,可以指定类别。(可以说是想要啥就有啥)。

--save-txt            save results to *.txt    #将预测的bounding box保存为txt文件
--save-crop           save cropped prediction boxes   #将预测的bounding box截取出来
 --classes CLASSES [CLASSES ...]    #只预测指定类别

4.使用的前提是,你要有你需要类别的权重文件(...pt),这个权重文件一般就用别人训练好的,或者就用yolo官方训练好的权重,然后用detect.py去预测你数据集的图片,就可以自动生成label,可以满足绝大部分需求,提高效率。

5.可能图中会遇到报错:

利用yolov5的预测结果,生成txt标注文件(与labelImg一样),实现自动标注_第2张图片

 6.解决办法:找到dataset.py的161行;

利用yolov5的预测结果,生成txt标注文件(与labelImg一样),实现自动标注_第3张图片

 7.重新开始运行。

利用yolov5的预测结果,生成txt标注文件(与labelImg一样),实现自动标注_第4张图片

8.如果数据量较大,请注释掉detect.py的186-203行。 

你可能感兴趣的:(深度学习,python,目标检测,计算机视觉,人工智能)