疫情期间佩戴口罩检测之训练检测口罩模型算法实现口罩检测步骤以及报错解决

最新推荐文章于 2023-03-23 16:47:26 发布

南蓬幽

最新推荐文章于 2023-03-23 16:47:26 发布

阅读量1.1k

点赞数 1

分类专栏：机器学习文章标签：算法 python 深度学习目标检测

本文链接：https://blog.csdn.net/qq_46556714/article/details/125291925

版权

机器学习专栏收录该内容

31 篇文章 5 订阅

订阅专栏

这篇博客介绍了如何训练一个口罩检测模型，使用的是YOLOX框架。首先，从GitHub上下载项目，然后安装环境，处理数据集，创建训练和测试样本的索引。在代码中调整参数，如设置类别数为2，更新类别名称。在训练过程中，可能会遇到路径、Dataloader、环境冲突等问题，通过修改sys.path、设置数据加载器的工作进程数和环境变量来解决。最后，启动训练模型时可能会出现错误，需要进一步调试相关文件。

摘要由CSDN通过智能技术生成

训练检测口罩模型

下载项目

https://github.com/Megvii-BaseDetection/YOLOX
在这里插入图片描述
打开项目

安装环境

在这里插入图片描述

pip install -r requirements.txt

在这里插入图片描述
如果报错

pip install -c esri pycocotools

第一步下载数据集

在datasets文件夹下新建VOCdevkit文件夹并将下载的data文件夹放进去，并改名为VOC2020。
在VOC2020文件夹下新建ImageSets文件夹，在建Main文件夹。

在这里插入图片描述
下载make_train_and_test.py文件并运行

import os
import random
 
train_percent = 0.8
test_percent = 0.2
xmlfilepath = 'VOCdevkit/VOC2020/Annotations'
txtsavepath = 'VOCdevkit/VOC2020/ImageSets'
total_xml = os.listdir(xmlfilepath)
 
num = len(total_xml)
list = range(num)
tr = int(num * train_percent)
te = int(num * test_percent)
train = random.sample(list, tr)
test = random.sample(list, te)
 
ftest = open('VOCdevkit/VOC2020/ImageSets/Main/test.txt', 'w')
ftrain = open('VOCdevkit/VOC2020/ImageSets/Main/train.txt', 'w')

for i in list:
    name = total_xml[i][:-4] + '\n'
    if i in train:
        ftrain.write(name)
    else:
        ftest.write(name)
 
ftrain.close()
ftest.close()

生成训练样本和测试样本的下标
在这里插入图片描述

第二步修改代码

在这里插入图片描述
yolox_voc_s.py相关代码改为

self.num_classes = 2

在这里插入图片描述

然后

voc_classea.pu改为

VOC_CLASSES = (
    "no_masked",
    "masked",
)

voc.py修改

在这里插入图片描述

训练模型

python tools/train.py -f ./exps/example/yolox_voc/yolox_voc_s.py -d 1 -b 4 --fp 16 -o

会报错
Traceback (most recent call last):
File “./tools/train.py”, line 13, in
from yolox.core import launch
ModuleNotFoundError: No module named ‘yolox’

在这里插入图片描述
路径错误
在tools/train.py开头加上开头路径

import sys
sys.path.append(r"D:\PycharmProjects\YOLOX-main")

在这里插入图片描述

报错 DataLoader worker

在这里插入图片描述
yolox_voc_s.py添加

 self.data_num_workers = 0

在这里插入图片描述

报错OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.

环境冲突
tools/train.py添加代码

import os
os.environ['KMP_DUPLICATE_LIB_OK'] = 'True'

在这里插入图片描述

报错 ERROR | yolox.core.launch:98 - An error has been caught in function ‘launch’, process ‘MainProcess’ (22360), thread ‘MainThread’ (18700):

在这里插入图片描述
修改
voc.py