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

原创

小南蓬幽 2022-06-30 10:03:20 博主文章分类：机器学习 ©著作权

文章标签 xml sed 数据集 文章分类 深度学习人工智能 在51CTO的第一篇博文 阅读数133

训练检测口罩模型

https://github.com/Megvii-BaseDetection/YOLOX

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

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

pip install -r requirements.txt

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

pip install -c esri pycocotools

第一步下载数据集

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

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

下载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()

生成训练样本和测试样本的下标

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

第二步修改代码

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

yolox_voc_s.py相关代码改为

self.num_classes = 2

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

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

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

voc_classea.pu改为

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

voc.py修改

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

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 <module> from yolox.core import launch ModuleNotFoundError: No module named 'yolox'

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