1.使用环境cuda 10.0,环境装好后使用nvcc -V检测,出现以下说明cuda安装完成
2.下载资源,yolo3-pytorch口罩识别检测,路径为:https://download.csdn.net/download/liaoqingjian/20969434?spm=1001.2014.3001.5503
3.下载好后使用pycharm打开,里面包含口罩数据集和口罩模型
4.下载重要的torch1.2和torchvision0.4.0库,使用命令
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
5.其他所需要的的库都可以用pip install XXX 来下载,缺什么下什么,不会出现下载不了的情况
6.所有的都下载完成后,直接运行video.py文件来运行视频检测
7.运行效果展示
8.如果想重新训练,运行train.py文件
9.如果想图片处理,运行predict.py文件