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

本文介绍了如何运用Yolov5进行目标检测,包括下载代码、运行预测脚本和转换结果为TXT格式。通过指定类别权重文件,可以自动生成数据集的标注。在运行过程中可能会遇到数据集类目不匹配的问题,解决方法是修改dataset.py的相应行。若数据量大,建议注释掉detect.py的部分代码以优化效率。
摘要由CSDN通过智能技术生成

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行。 

  • 12
    点赞
  • 110
    收藏
    觉得还不错? 一键收藏
  • 11
    评论
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值