- 🍨 本文为🔗365天深度学习训练营 中的学习记录博客
- 🍖 原作者:K同学啊|接辅导、项目定制
本次主要是记录使用yolov5检测图片和视频的过程
地址为:https://github.com/ultralytics/yolov5
有两种方法
通过下述命令安装源码中给出的配置环境文件
pip install -r requirements.txt
我个人缺少一个CV2模块,通过下述命令进行安装
pip install opencv-python>=4.1.1 --trusted-host pypi.douban.com/simple/
也有多种方法进行运行,比如cmd、jupyter、Anaconda prompt
我这里使用的是Anaconda prompt
在激活你的环境后,输入下方代码:
python C:\Users\yolov5-master\detect.py --source C:\Users\Desktop\p.jpg --weights weights/yolov5s.pt --conf-thres 0.25
需要找到detect.py的路径以及待检测的图片的路径
第一次使用会下载权重,会慢一些
早上上班随手拍了一张路边的图片用于检测,其实还在下雪
完成后会显示存储路径:Results saved to \yolov5-master\runs\detect\exp
我们找到这张图片:
跟上方图片检测一样,切换文件路径即可
python C:\Users\yolov5-master\detect.py --source C:\Users\Desktop\v1.mp4 --weights weights/yolov5s.pt
检测了110次
一样是随手拍的视频
检测前
v1
检测后生成文件
v1-y
1.本来想用jupyter运行,因为源码中有tutorial文件,是有运行代码的,但是一直卡住不动
2.cmd启动,输完命令敲回车没有任何反应,也没报错,直接换到下一行,不怕报错,就怕没错去改。。
最终还是选用了Anaconda prompt,一次成功