（19）目标检测算法之模型预测并保存结果到xml中，自动生成标签文件/自动标注

目标检测算法之模型预测并保存结果到xml中，自动生成标签文件/自动标注

• 在做深度学习算法研究中，标图往往是工作量很大且性价比较低的一项工作了，最近在用YOLO，在此参考网上已有程序完善下可执行demo
• 看过一下demo，需要导入utils下的模块，我没有安装成功，在此特别感谢这篇文章：YOLOv5实现半标注—告别大量重复标注工作，本文是在其代码上做了些更改

1. 自动生成标签实现

• 具体过程的话呢需要一些几点：

  • 1. 对少些图像进行标注并训练，保存模型权重文件
  • 2. 封装预测模块，并将预测结果整理成代码中所需要的格式：由[x_min,y_min,x_max,y_max,label[x_min,y_min,x_max,y_max,label]组成的列表
  • 3. 修改测试路径，运行自动检测demo，批量生成标签

• 我用的是YOLO，如果你的是其他模型，自行做出相应修改