目录

介绍

从带有LabelImg的图像生成训练数据集

下一步


在这里,我们将看到如何正确标记结果图像以训练YOLO模型。

介绍

在本系列的上一篇文章中,我们为AI模型训练准备了口罩图像。在这一篇中,我们将标记这些图像。如果您对计算机视觉任务不是很熟悉,则可能没有听说过LabelImg。本质上,它是用于计算机视觉的图像标记工具。还有其他功能——但此功能非常易于使用、轻巧,与YOLOv5兼容且免费!

请放心,因为此过程将花费数小时。

要安装该应用程序,您可以使用多个存储库。使用git clone克隆这个仓库。在同一终端窗口中,运行以下命令进行安装:

Ubuntu:

cd path/to/dir/LabelImg
sudo apt-get install pyqt5-dev-tools
sudo pip3 install -r requirements/requirements-linux-python3.txt
make qt5py3

MacOS(首先导航到LabelImg目录)

brew install qt
pip3 install pyqt5 lxml
make qt5py3

Windows + Anaconda(首先导航到LabelImg目录):

conda install pyqt=5
conda install -c anaconda lxml
pyrcc5 -o libs/resources.py resources.qrc

安装完成后,您需要编辑预定义的自定义类。导航到labelImg/data目录,然后搜索预定义_classes.txt文件。打开它,然后输入要在应用程序本身中使用的标签。对于我们的口罩检测项目,这是它们的外观:

现在,通过python3 labelImg.py发出启动应用程序。该应用程序将启动:

现在,让我们动手使用图像标签。它需要很多耐心,但是您可以做到!

从带有LabelImg的图像生成训练数据集

在启动标签过程之前,请打开文件资源管理器并为您放大后要保存的图像创建一个文件夹。将您在该数据扩充步骤中获得的文件解压缩到该目录中,并在其中创建一个名为Labels的文件夹,我们将在其中保存LabelImg生成的所有文本文件。

是时候标记图像了。打开LabelImg,单击左侧面板上的Open Dir,然后选择包含图像数据集的文件夹。要选择标签的存储位置,请点击更改保存目录,然后选择在上面几行中创建的新标签文件夹。完成后,您将看到该文件夹​​的第一张图片:

在开始标记过程之前,请确保在左侧面板上选择了“YOLO”模式:

单击“创建矩形框然后在图像的每个面周围绘制一个正方形。尝试尽可能多地遮盖每个面孔,同时不要在正方形内留有较大的空白空间。每次绘制正方形时,都会出现一个弹出窗口,提示您选择与正方形相对应的标签:

在该过程的最后,您应该获得一张完全标记的图像:

请注意右侧的标签摘要。用它来跟踪您对每个图像所做的操作。准备就绪后,点击左侧面板上的保存,然后点击下一步图像

经过数小时的娱乐之后,您为包含4,000多个图像的数据集添加了标签。生成的g .txt文件包含什么?让我们检查其中之一:

每个.txt文件都链接到其相应的图像。例如,如果有一个名为0.jpg的图像,则LabelImg将生成一个名为0.txt的文件,其中包含图像中每个标签的坐标。对于每个对象,都会创建一个新行,其格式为<object class number> <x> <y> <width> <height>。这称为YOLO标签格式。

下一步

标记数据集后,最重要的部分的时机已经成熟。在接下来的文章中,我们将使用数据集来训练YOLOv5对象检测模型。敬请关注!

https://www.codeproject.com/Articles/5293072/Labelling-Images-for-AI-Model-Training

(三)标记图像以进行AI模型训练相关推荐

  1. ONNX系列三 --- 使用ONNX使PyTorch AI模型可移植

    目录 PyTorch简介 导入转换器 快速浏览模型 将PyTorch模型转换为ONNX 摘要和后续步骤 参考文献 下载源547.1 KB 系列文章列表如下: ONNX系列一 --- 带有ONNX的便携 ...

  2. InfoWorld文章丨将数据编排技术用于AI模型训练

    This article was originally published on InfoWorld on March 22, 2022. Reprinted with permission. IDG ...

  3. AI模型训练、实施工程师的职业前景怎么样?

    本篇文章主要讲解ai模型训练.模型实施工程师的职业前景和趋势分析 作者:任聪聪 日期:2023年4月18日 ai训练师.模型实施工程师,一般是指opencv.pytorh.python.java.机械 ...

  4. 【AMD GPU】使用A卡进行ai模型训练

    在linux中使用A卡进行ai模型训练 吐槽 使用的设备配置 2022-10-24 23:21:50一键部署工具发布 安装GPU驱动 添加path 安装MIopen RDNA2架构安装pytorch ...

  5. 药盒识别/垃圾分类—高精度AI模型训练及边缘部署分享

    这是AI应用的时代. 工厂:高速照相机拍下传送带上的零部件,机器识别出有缺陷的零件剔除: 农场:拍摄虫体照片,识别害虫名称和数量并能防控虫害,减少农药使用量,发展绿色农业 小区:智能垃圾箱可以基于图像 ...

  6. 【华为云技术分享】网络场景AI模型训练效率实践

    问题 在网络场景下的AI模型训练的过程中,KPI异常检测项目需要对设备内多模块.多类型数据,并根据波形以及异常表现进行分析,这样的数据量往往较大,对内存和性能要求较高. 同时,在设计优化算法时,需要快 ...

  7. AI 绘画(2):Ai模型训练,Embedding模型,实现“人物模型“自由

    文章目录 文章回顾 感谢人员 题外话 Ai绘画公约 Ai模型训练 硬件要求 显存设置 查看显存大小 显存过小解决方法 视频教程 前期准备 SD配置设置 SD设置配置 SD训练配置pt生成 训练集收集 ...

  8. 大三学生独自破解逆天AI模型:我只是把撩妹的时间,都用来研究机器学习了...

    问耕 发自 雁栖湖 量子位 出品 | 公众号 QbitAI GPT-2,一个逆天的AI. 今年2月,OpenAI首次对外公布了这个模型的存在.GPT-2写起文章来文思泉涌毫无违和感,无需针对性训练就能 ...

  9. 如何借助分布式存储 JuiceFS 加速 AI 模型训练

    传统的机器学习模型,数据集比较小,模型的算法也比较简单,使用单机存储,或者本地硬盘就足够了,像 JuiceFS 这样的分布式存储并不是必需品. 随着近几年深度学习的蓬勃发展,越来越多的团队开始遇到了单 ...

最新文章

  1. Swift3.0带来的变化汇总
  2. python词频作图_基于Python的词频分析与云图生成
  3. python内置函数open_python_内置函数和open
  4. python操作系统-PYTHON-操作系统基础
  5. 将NetConf转成NormalizedNode对象及NormalizedNode对象转回NetConf(序列化与反序列化)
  6. 填表2018-11-11
  7. PHP——smarty模板(第一天)
  8. 牛客 - 做计数(数学)
  9. Java 8 新增lambda表达式(-)
  10. 随想录(中间件接口的定义方法)
  11. Zookeeper脑图
  12. 【心电信号】基于matlab小波变换心电信号去噪【含Matlab源码 956期】
  13. 教你电脑系统如何深度清理c盘空间
  14. ASP的工作原理和运行环境
  15. H3C交换机常用命令大全
  16. git错误:unable to auto-detect email address
  17. ##24点游戏程序实现算法(文末源代码)
  18. 易知微当选中国互联网协会数字孪生技术应用工作委员会成员单位
  19. 【JAVA EE#6】【传智书城·源码阅读】后台管理模块:权限控制+页面分析+商品管理+销售榜单+订单管理+公告管理+项目结构思维导图
  20. 产品开发中项目与项目管理

热门文章

  1. matlab与水库调度,蛙跳算法优化水库调度,全局迭代中最优解未更新
  2. linux网速卡怎么办_手机同是4G网络,为什么总比别人慢,学会这个方法,网速成倍提升...
  3. 专供PNG免抠设计素材好地方,做设计到搜图114
  4. 配置的android版本,Android SDK配置(V3.0.0及以上版本)
  5. plc 上位机编译算法_什么是PLC与DDC PLC与DDC的区别
  6. PCI Express®Basics Background | PDF
  7. Linux如何处理 Too many open files
  8. Linux内核:Gigantic巨页与CMA的结合的PATCH补丁提交
  9. 软件开源是如何赚钱?
  10. Mellanox能给英伟达带来什么?