本文为为🔗365天深度学习训练营内部文章
原作者:K同学啊
一 安装源码
开源网址:GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
二 安装所需环境
安装环境依赖包,进入项目文件夹在文件地址栏中直接输入cmd,打开cmd后输入:pip install -r requirements.txt
三 运行代码
以同样的方法打开cmd,输入命令:python path/to/dectect.py --source path/to/img.jpg --weights yolov5s.pt --img 640,注意需要将图片路径替换为自己的,例如:python detect.py --source data\images\zidane.jpg --weights yolov5s.pt --img 640。这里使用的是官方已经训练好的模型。
进入 runs\detect\exp2查看结果
再来一例:
四 视频检测
同样的方法打开cmd输入:python detect.py --source data\videos\MyVideo_2.mp4
会得到如下结果: