YOLOv5：调用官方权重进行检测

- **🍨 本文为[🔗365天深度学习训练营](https://mp.weixin.qq.com/s/rnFa-IeY93EpjVu0yzzjkw) 中的学习记录博客**
- **🍖 原作者：[K同学啊](https://mtyjkh.blog.csdn.net/)**

一：前言

本篇以Yolov5s为例，官方开源地址为：https://github.com/ultralytics/yolov5

二：下载源码

安装需要的环境

• 安装环境依赖包，windows请进入项目文件在文件地址栏中直接输入cmd，打开cmd后输入：pip install -r requirements.txt
• Mac 则在项目文件的终端打开

Windows 版本：

Mac 版本：

三：运行代码

以同样的方法打开cmd或终端，输入命令：python path/to/detect.py --source path/to/img.jpg --weights yolov5s.pt --img 640,注意需要将图片路径替换成你的，例如我的是：python detect.py --source /Users/brandom/Downloads/f9b6b3e0-226f-11f0-b1bf-e2875d08fa6c.jpeg --weights yolov5s.pt --img 640 。这里我们使用的是官方已经帮我们训练好的模型。

命令运行成功后，你将会得到：

进入YOLO的文件夹查看：

得到我们的检测结果：

四：视频检测

同样的方法打开cmd或终端输入：python detect.py --source MyVideo_2.mp4

得到如下结果：