安装YOLOv5模型环境并测试

安装YOLOv5模型环境并测试

下载yolov5源代码

YOLOv5 模型源码:https://github.com/ultralytics/yolov5

配置环境

下载后可以在文件夹中看到requirements.txt文件,其中所给的就是YOLOv5模型所需要的环境配置。因为跑不同的模型所需要的环境可能不同,在此建议大家可以使用虚拟环境来进行配置。
详细步骤见我之前的博客:win10下载Anaconda并创建虚拟环境安装pytorch
设置好虚拟环境后(或者你就在本机)打开Anaconda Prompt,如果设置了虚拟环境就先activate 你的环境名进入。
使用 cd 进入到YOLOv5文件夹,来到requirements.txt 所在的路径下,输入

pip install -U -r requirements.txt

大部分的环境用这条语句都是可以配置上的,大概会卡在torch的下载上,大家同样可以参考我之前的那篇博客win10下载Anaconda并创建虚拟环境安装pytorch,自己手动安装torch和torchvision。

测试环境

如果环境都安装完成了,下面我们测试一下YOLOv5的模型是否可以成功运行,打开spyder(如果你设置了虚拟环境,就打开带有你虚拟环境名后缀的spyder)。运行detect.py文件。这是在用他本来跑好的模型来进行测试,测试的图片在yolov5/inference/images里面,可以自己手动加,这里我就自己加了几张图片,原本的文件中应该只有自带的两张图片。如果可以看到如下的情况,并在yolov5/inference/output文件夹中看到标注好的文件,那么恭喜你,你的YOLOv5模型环境配置完成啦!
在这里插入图片描述
参考:
https://blog.csdn.net/laovife/article/details/106802725?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.channel_param&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.channel_param

https://blog.csdn.net/ai_faker/article/details/107099907

最后附上我在安装过程中帮助比较大的两篇文章,我在安装的过程中遇到了很多的问题,在此没有一一列举出来,如果大家遇到了问题欢迎给我留言交流,希望这篇文章可以帮助到大家!后续我会持续更新使用YOLOv5模型跑自己的数据集以及参数的调整。

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值