labelGo:基于 YOLOv5 的辅助标注工具

文章介绍了一个基于YOLOv5的自动标注工具,用于深度学习检测项目的数据预处理。用户可以克隆项目,配置环境,修改类别文件,然后使用该工具自动标注图片,提高标注效率。此外,还提到可以利用初步标注的数据训练模型,进一步优化自动标注过程。
摘要由CSDN通过智能技术生成

目录

1.从 git 中获取项目

2.切换操作目录到工程目录

3.配置环境

4.将目录中的/data/predefined_classes.txt文件的内容修改为你自己的类别

5.启动应用程序

6.点击“打开目录”按钮选择存放图片的文件夹

7.点击“自动标注”按钮确认信息无误后选择已训练的yolov5 pytorch模型完成自动标注

8.根据实际要求,对自动标注结果进行调整后保存即可


在我们平常做 深度学习检测项目时,可能经常需要标注数据,但是标注数据,本身又是一件很麻烦的事情。今天,看到 Github 有个大佬开源的基于 YOLOv5 的辅助标注工具,推荐给大家。

首先,上链接:

https://github.com/cnyvfang/labelGo-Yolov5AutoLabelImg

这是一个基于 labelImg与 YOLOv5 的半自动标注工具,通过现有的YOLOv5 PyTorch模型对数据集进行半自动标注。

它的原理其实很简单,使用在 COCO 数据集训练好的 YOLOv5 模型,在你的数据集上进行推理,生成标签。

      大家看一下官方给的两个动图,应该就明白怎么使用了。

bddb460e5edcbf28c0297e03f7285782.png

ba9da56b9146bbbaa958b98b2231e975.gif

其实,我们也可以自己先标注一部分数据,训练一个基本的模型,然后使用跟这个模型帮我们标注,提高标注效率。有点半监督的意思了。

官方给出的具体使用方法如下:

1.从 git 中获取项目

git clone https://github.com/cnyvfang/labelGo-Yolov5AutoLabelImg.git

2.切换操作目录到工程目录

cd labelGo-Yolov5AutoLabelImg

3.配置环境

pip install -r requirements.txt

4.将目录中的/data/predefined_classes.txt文件的内容修改为你自己的类别

(可选).变更成中文版,请编辑\libs\stringBundle.py,注释第52行,取消注释54行。

5.启动应用程序

python labelGo.py

6.点击“打开目录”按钮选择存放图片的文件夹

7.点击“自动标注”按钮确认信息无误后选择已训练的yolov5 pytorch模型完成自动标注

8.根据实际要求,对自动标注结果进行调整后保存即可

参考引自:

labelGo:基于 YOLOv5 的辅助标注工具https://mp.weixin.qq.com/s?__biz=MzI2Nzg4NjA5OQ==&mid=2247512453&idx=2&sn=54aa8e7df44d5e906eb2b9d22c3b6f61&chksm=eafafaf9dd8d73ef519e0f35967d7dbaac9dcd5792b4d850655748d9ea7f0ef1a17ce87d66a7&scene=126&&sessionid=0

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值