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