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的种类全部写进去。


AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。
更多推荐



所有评论(0)