一、大致流程

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

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

 

 

Logo

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

更多推荐