深度学习算法在目标识别分类任务上取得了飞快的进展,也取得了优异的效果,本文将重点介绍如何运用YOLOv8模型进行目标分类检测任务,以口罩的分类为例。
0.环境准备:首先需要安装配置好深度学习相关的环境。
1.模型的搭建:模型框架源于Github网站,模型的代码层次如图所示。
2.数据集的准备(标注):为了制作合适的口罩分类识别的数据集,先从网上寻找一部分开源的数据集,并根据程序需要进行部分修改,然后通过自己爬虫采取的方式寻找到一部分合适的图片,使用 Labelimg工具对其进行标注,标注完成后会生成与图片对应的txt文档,这里要保证图片与txt文档一 一对应,标注过程如图所示。
数据集中各类别图片数量:
图片种类 |
数量 |
all cloth(布口罩) kn95 n95 surgical nomask |
2606 740 422 |