yolov5自动标注数据集,保存txt--(yolo格式的标签)

准备工作:

a、先个人使用labelimg或者其他标注软件标注一部分数据集
b、训练一个有一定识别能力的yolov5模型
c、先使用模型推理一遍,看效果如何,是否能框上很多目标
d、调整置信度--conf-thres的大小,例如0.10.20.250.45这类的常用数值,觉得检测效果不错了就可以开始下面的工作了

1、找到yolov5的推理文件detect.py,打开保存txt开关,如图:
在这里插入图片描述
这个开关是保存标签的,会将输入的图片经过推理,并将图片和标签一同保存在runs/detect/exp/下面,如图片:
在这里插入图片描述
2、在detect.py的163-177行代码,如下图,把图片中的部分代码注解掉,这样就不会在图片上或者结果的框,而是直接复制过去,并保存标签。(或者你能到生成的标签,推理后的图片就不要了,直接和输入的源图片一起作为数据集,这样也是可以的。)
在这里插入图片描述

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值