Windows搭建YOLOv5-P6 models

一、YOLOV5官方发布版本

GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

二、安装python环境库

1、新建虚拟环境

conda create -n yolov5 python=3.8

2、定位到代码目录下安装环境库

设置镜像加速
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
cd yolov5
pip install -r requirements.txt  # install

三、下载模型的预训练模型

YOLOV5官方发布模型

四、测试环境搭建成功

python path/to/detect.py --weights yolov5s.pt --source 0         # webcam，读取摄像头视频流
                                                      path/to/img.jpg    # image，读取图片
                                                      vid.mp4        # video,读取视频
                                                      path/          # directory，读取文件
                                                      path/*.jpg     # glob
                                                      'https://youtu.be/Zgi9g1ksQHc'  # YouTube
                                                      'rtsp://example.com/media.mp4'  # RTSP, RTMP, HTTP stream

运行视频流
python detect.py --weights yolov5s.pt --source 0