Skip to content

QikaiXu/Mask-Wearing-Detection

Repository files navigation

口罩佩戴检测

浙江大学《人工智能与系统》课程作业,口罩佩戴检测。

项目来源于:https://mo.zju.edu.cn/workspace/5f747601dfcc16df6114a426?type=app&tab=2(不确定能不能访问)。

项目介绍可以看 torch_main.ipynb

数据集下载:datasets。放在蓝奏云上,域名可能经常会换,如果访问不了了,就更换一下前面的域名。用链接最后的一串 id 就能定位到。

这里修改了 torch_py/MobileNetV1.py 中网络的结构。

训练和推理代码在 torch_train.ipynbtrain.py 中,其中 torch_train.ipynb 取代价最低的模型,而 train.py 会取验证集准确率最高的模型,效率会降低。

About

口罩佩戴检查,Pytorch,CV,CNN。

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published