说明:
如果yolov5的预训练权重已经包括了你所需要的类别,可直接进行自动标注,无需自行训练模型。
步骤:
1.安装与环境配置yolov5-6.0版本
yolov5-6.0 github链接:https://github.com/ultralytics/yolov5/tree/v6.0
2.标注数据进行模型训练
使用labelme进行标注
使用yolov5-6.0模型训练
3.使用训练好的模型进行自动标注
标注工具代码链接:https://github.com/mohenghui/detectAuto
4.xml转txt 标签格式转换:
5.去除未标注图像
条件:txt一定少于图像