一、前言
Yolov5四种网络模型
Yolov5官方代码中,给出的目标检测网络中一共有4个版本,分别是Yolov5s、Yolov5m、Yolov5l、Yolov5x四个模型。
2. 下载源码
5s:https://github.com/ultralytics/yolov5
安装:
Clone repo and install requirements.txt in a Python>=3.7.0 environment, including PyTorch>=1.7.
git clone https://github.com/ultralytics/yolov5 # clone
cd yolov5
pip install -r requirements.txt # install
3. 运行代码
python detect.py --source data\images\zidane.jpg --weights yolov5
s.pt --img 640
4. 视频检测
python detect.py --source data/images/mar1.mp4