栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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

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

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/768997.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号