yolov5自动打标签
时间: 2023-10-15 20:06:44 浏览: 162
全自动标注工具支持YOLOv5,真正解放双手!
3星 · 编辑精心推荐
yolov5可以通过使用auto_labelimg.py实现自动标注功能。具体步骤如下:
1. 首先使用yolov5训练一个小批量样本(例如100个),得到.pt文件。
2. 在yolov5文件夹中放置auto_labelimg.py文件。
3. 修改auto_labelimg.py文件中的第176和178行,将weights文件路径修改为之前训练得到的.pt文件路径,将source修改为需要自动标注的图片文件夹位置。
4. 运行auto_labelimg.py文件,会在图片文件夹外部生成annotations文件夹,其中包含自动生成的.xml文件。
5. 可以使用labelImg软件对自动生成的新.xml文件进行调整。
阅读全文