YOLOv5实战口罩识别项目

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
YOLOv5实战口罩识别项目_第1张图片
使用pycharm打开并且配置解释器
YOLOv5实战口罩识别项目_第2张图片
将 yolov5s.pt,yolov5m.pt,yolov5l.pt,yolov5x.pt权重文件(在资源包里),并放置在weights文件夹下
YOLOv5实战口罩识别项目_第3张图片

安装测试

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

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

![在这里插入图片描述](https://img-blog.csdnimg.cn/badd4c40acd44ba6a7eeccd35a6603e6.png
YOLOv5实战口罩识别项目_第4张图片
YOLOv5实战口罩识别项目_第5张图片

你可能感兴趣的:(口罩识别,pycharm,python,深度学习)