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

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

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

2.pycharm终端运行:

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


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.可能图中会遇到报错:

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

 7.重新开始运行。

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

欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/langs/789622.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-05
下一篇 2022-05-05

发表评论

登录后才能评论

评论列表(0条)

保存