YOLO 自动标注工具autolabelimg（适用yolov5 v6.1）

一、获取自定义数据集的YOLOv5.pt权重

   首先，手动标注一部分数据集，训练YOLOV5 模型，得到best.pt 。

二、下载和安装autolabelimg 

   （1） github链接：  https://github.com/wufan-tb/AutoLabelImg.git

    手动下载.zip  ,也可以 clone：

git clone https://github.com/wufan-tb/AutoLabelImg.git

（2）安装 

创建虚拟环境-----激活-----安装，安装命令如下：

pip install –r requirements.txt -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

三、  使用

（1）  切换到虚拟环境下，输入以下命令：

python labelimg.py

（2） best.pt 放在指定路径下（重要！）

将训练好的best.py 放在 pytorch_yolov5/weights/下。

（3）具体步骤：

思路：  首先手动标注一部分数据集，然后训练出一个yolov5 模型，获取 best.pt ，然后用autolabelimg软件加载best.pt,进行自动标注。 然后人工再进行标注调整。

原理：使用yolov5对图片进行predcit,将结果中的矩形框信息通过labelimg显示出来。

参考资料：Autolabelimg自动标注工具_labelimg的原理-CSDN博客