YOLOv7半自动标注

一、大致流程

对于每一个目标识别方向的萌新来说,数据都是最重要的一部分,一个好的数据集可以对我们训练结果产生非常好的结果。但是,庞大的数据集标注却占用了我们非常多的时间,因此对于目标识别方向的我们,使用半自动标注方式将会节省我们大量的时间。

使用YOLOv5也是可以进行半自动标注,这里我没实验过。YOLOv7成功对我们的数据进行标注。由于最终得到的是txt文件,所以这里标注软件使用的是labelimg。当然大家也可以尝试下将txt文件转成xml或者json。

大致流程:标注部分数据集-->使用YOLOv7进行训练-->得到best.pt权重文件。而我们就是要使用best.pt文件去标注我们自己剩余的数据集。

二、操作步骤

 1、save--text 这行命令的意思是是否保存使用预训练权重生成的txt文件,默认不保存,我们需要将

action='store_true'改成action='store_false'

2、进行预训练,得到run/detect/exp文件

这里的labels是我们训练得到的txt文件,分为类别、x、y、w、h

 3、 使用labelimg打开照片,然后导入txt文件就OK了。

这里需要注意的是,我们需要手动添加一个类别,不然识别不出来。例如CoCo数据集,那么我们需要把COCO的种类全部写进去。

 

 

  • 1
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值