文件名称:基于PyTorch的口罩检测与分类mask_detection代码详解
文件大小:49.38MB
文件格式:ZIP
更新时间:2023-11-15 08:53:11
口罩检测;人脸检测; 口罩识别 Covid-19 检测是否戴口罩 CNN,人脸检测
针对口罩检测,进行调研,包括了分类器的设计,人脸检测的设计。 其中主要分为两阶段,先是人脸检测,然后将检测到的人脸,进行二分类,标签为戴口罩,不戴口罩。包括了pytorch的代码,ppt的讲解,技术文档。 人脸识别是MTCNN,然后分类可以自己训练,我采用的是ResNet-18.
【文件预览】:
classifier.py
original_model
----rnet_epoch.pt(237KB)
----anno_train.txt(3.59MB)
----pnet_epoch.pt(29KB)
----onet_epoch.pt(879KB)
finetuned_model_resnet18.pth
mtcnn
----config.py(1KB)
----train_net()
--------train_r_net.py(3KB)
--------train_o_net.py(953B)
--------train.py(9KB)
--------__pycache__()
--------__init__.py(0B)
--------train_p_net.py(3KB)
----__pycache__()
--------__init__.cpython-37.pyc(176B)
--------__init__.cpython-35.pyc(155B)
--------config.cpython-35.pyc(1KB)
----data_preprocess()
--------gen_landmark_48.py(5KB)
--------assemble_onet_imglist.py(687B)
--------gen_Rnet_train_data.py(8KB)
--------gen_Pnet_train_data.py(5KB)
--------utils.py(2KB)
--------__pycache__()
--------assemble_pnet_imglist.py(689B)
--------assemble.py(1KB)
--------gen_Onet_train_data.py(7KB)
--------assemble_rnet_imglist.py(689B)
----__init__.py(0B)
----core()
--------models.py(9KB)
--------resnet_inception_v2.py(9KB)
--------image_tools.py(1KB)
--------utils.py(4KB)
--------__pycache__()
--------nms.py(1KB)
--------__init__.py(0B)
--------detect.py(25KB)
--------imagedb.py(5KB)
--------vision.py(5KB)
--------image_reader.py(4KB)
04.jpg
leslie.jpg
s_l.jpg
l_z.png
.ipynb_checkpoints
----mask_classifier-checkpoint.ipynb(477KB)
----face_detection-checkpoint.ipynb(72B)
mask_classifier.ipynb
model_store
----rnet_epoch.pt(237KB)
----pnet_epoch.pt(29KB)
----onet_epoch.pt(879KB)
----tips.txt(75B)
----__init__.py(0B)
r_4.jpg
anno_store
----pos_12.txt(0B)
----neg_24.txt(0B)
----pos_48.txt(0B)
----neg_48.txt(0B)
----landmark_48.txt(0B)
----part_48.txt(0B)
----anno_train.txt(0B)
----imglist_anno_48.txt(2.9MB)
----part_12.txt(0B)
----tool()
--------anno_train.txt(53KB)
--------anno_train_fixed.txt(54KB)
--------change.py(775B)
--------tips.txt(243B)
--------format()
----imglist_anno_12.txt(0B)
----part_24.txt(0B)
----neg_12.txt(0B)
----tip.txt(40B)
----__init__.py(0B)
----imglist_anno_24.txt(0B)
----pos_24.txt(0B)
results
----r_1.jpg(53KB)
----r_4.jpg(75KB)
----r_3.jpg(85KB)
----r_2.jpg(79KB)
mask_detection讲解文档.zip
data
----data_read.md(201B)
mtcnn_test.py
result_new.jpg