YOLOv5实战口罩识别系统

新建虚拟环境

conda create -n yolo_v5 python=3.7

切换虚拟环境

source activate yolo_v5

安装pytorch

pip3 install torch torchvision torchaudio

克隆yolov5项目

https://github.com/ultralytics/yolov5

使用pycharm打开并且配置解释器

将 yolov5s.pt，yolov5m.pt，yolov5l.pt，yolov5x.pt权重文件(在资源包里)，并放置在weights文件夹下

安装测试

测试图片：
在yolov5路径下执行

python detect.py --source ./data/images/ --weights weights/yolov5s.pt --conf 0.4

![在这里插入图片描述](https://img-blog.csdnimg.cn/badd4c40acd44ba6a7eeccd35a6603e6.png