首页 > 其他分享 >YOLO 自动标注工具autolabelimg(适用yolov5 v6.1)

YOLO 自动标注工具autolabelimg(适用yolov5 v6.1)

时间:2024-09-10 13:51:49浏览次数:9  
标签:yolov5 v6.1 pt YOLO autolabelimg com best 标注

一、获取自定义数据集的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博客

标签:yolov5,v6.1,pt,YOLO,autolabelimg,com,best,标注
From: https://blog.csdn.net/m0_58054476/article/details/142096433

相关文章