深度学习Y1周：调用官方权重进行检测

• 🍨 本文为🔗365天深度学习训练营中的学习记录博客
• 🍖 原作者：K同学啊

学习的整体安排：

①跑通官方代码

②训练自己的数据集

③代码解析

④理论知识学习

学习预期：可以独立进行代码优化实验

一、前言

YOLO系列：目前最热门的目标检测算法

YOLO目前已经更新到了YOLOv12

TensorFlow与Pytorch区别：

• TensorFlow：更倾向于工程人员，落地比较好
• Pytorch：比较倾向于研究人员，前言算法用的比较多

YOLOv5分为YOLOv5s、YOLOv5m、YOLOv5l、YOLOv5X四个版本

本学习用YOLOv5s

二、下载源码

1.安装需要的环境

安装环境依赖包：进入项目文件在文件地址栏中直接输入“cmd”，打开“cmd”后输入：pip install -r requirements.txt

如何打开cmd命令行：

操作截图： 

三、运行代码

需要的环境：

python>=3.6.0

pytorch>=1.7

所有数据：

来自训练营

以同样方法打开“cmd”，输入命令：python path/to/detect.py --source path/to/img.jpg --weights yolov5s.pt --img 640

【注】需要将图片路径替换成自己的，使用的是官方已经帮忙训练好的模型

如：python detect.py --source data\images\zidane.jpg --weights yolov5s.pt --img 640

操作截图： 

 到运行文件夹中查看检测结果：

四、视频检测

同样方式打开“cmd”输入：python detect.py --source data\MyVideo_2.mp4

得到如下结果：

五、总结

1.在第三部分的时候可能遇到两个问题，

第一，TimeoutError：[WinError 10060] 由于连接方在一段时间后没有正确答复或者连接的主机没有反应，连接尝试失败。

第二，TypeError：unlink（） got an unexpected keyword argument 'missing_ok'

原因：网络问题，缺少yolov5s.pt文件

解决方法：多试几次/将yolov5s.pt放到项目主目录下就可以

2.YOLO作为广泛应用的算法，值得好好学习。