第Y1周打卡——调用官方权重进行检测

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

一、下载源码

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

​

二、安装环境

解压zip文件后，在文件目录内，打开cmd后输入：pip install -r requirements.txt

三、图片识别

python detect.py --source data\R.jpg --weights yolov5s.pt --img 640

四、视频识别

python detect.py --source MyVideo_2.mp4

总结：

学习收获

1. 源码理解与环境搭建：通过从GitHub下载YOLOv5源代码，并成功配置运行环境，我对PyTorch框架有了更深的理解。在解决依赖项安装过程中的各种问题时，也锻炼了我的问题排查能力。

2. 图片和视频识别实战：亲手执行图像和视频的目标检测任务，使我对YOLOv5的实际应用有了直观的感受。特别是看到模型能够准确地识别出复杂的场景中的物体，极大地激发了我的学习兴趣。

3. 技术文档阅读能力提升：在学习过程中，频繁参考Ultralytics官方文档和其他在线资源，我的技术文档阅读能力和快速学习新技能的能力得到了提升。

遇到的挑战与解决方案

• 环境配置难题：最初在尝试安装requirements.txt中的依赖项时遇到了一些兼容性问题。通过查阅社区讨论、更新Python版本以及创建新的虚拟环境，最终解决了这些问题。
• 模型优化困惑：在探索不同版本的模型（如yolov5s.pt, yolov5m.pt等）时，对于如何选择适合特定应用场景的模型感到迷茫。经过多次实验对比不同模型的表现，逐渐积累了经验。