目标检测第5步:如何在Windows 10系统下,搭建YOLOv5(5.0)环境?保姆级,没有人比这个更详细了(更新时间2022.3.22)

(请先看置顶博文)本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客

目录

一、安装Anaconda或Miniconda

二、安装CUDA

三、安装cuDNN

四、安装Python

五、安装Yolov5环境

1、下载Yolov5 5.0源码压缩包

2、解压并打开requirements.txt

3、安装Pytorch

4、安装其他依赖项

5、下载权重

六、运行Yolov5

附录一 下载安装Pytorch

附录二 下载安装torchversion


 这已经是目标检测第5步了,本篇文章的前四章将会对目标检测前4步进行一个回顾。

一、安装Anaconda或Miniconda

Miniconda安装步骤见博文:目标检测第1步:如何在Windows 10下安装MiniConda?_liO_Oil的博客-CSDN博客_miniconda

Anaconda安装步骤详见博文:目标检测第2步:如何在Windows 10下安装Anaconda?_liO_Oil的博客-CSDN博客_anaconda下载

       虽然有的同学就是想单纯试验一下Yolov5,但是按照这个教程安装Anaconda后,可以较为顺利的安装依赖,与此同时还为以后的代码运行做铺垫(因为Jupyter lab包含在其中,并且软件源已经更换了)

二、安装CUDA

CUDA的具体安装教程可以查看此博文:目标检测第3步:如何在Windows 10系统下安装CUDA(更新时间2022.03.22)_liO_Oil的博客-CSDN博客

三、安装cuDNN

cuDNN的安装步骤见博文:目标检测第4步:显卡、GPU、CUDA、cuDNN的介绍及如何在Windows 10下安装cuDNN?_liO_Oil的博客-CSDN博客

四、安装Python

Yolov5(5.0)需要Python3.7版本以上,所以这里提供一种安装Python的方法,Python的版本不要太高,我用的是3.9.7,详见博文:一、Pytho第一课——Python安装及配置路径方法(最详细小白教程,没有之一。如若不懂,不是还可以私信嘛!对吧?)_Cat-CSDN博客

五、安装Yolov5环境

1、下载Yolov5 5.0源码压缩包

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

打开网页后,如图1所示。点击Tags后选择5.0版本,并点击zip下载,如图2所示

图1 YOLOv5GitHub页面 

图2 下载Yolov5源码压缩包 

 学会了5.0环境的配置,6.0的不就无师自通了嘛?                                              

2、解压并打开requirements.txt

第一行“pip install -r requirements.txt”就是安装以下依赖包的命令,如图3所示.

pip install -r requirements.txt

 这个是卸载命令:

pip uninstall -r requirements.txt  #卸载requirements.txt文件里的所有依赖pip uninstall torch    # 卸载torch的命令

输入该安装命令后可实现“一键安装”的效果。因为是Python语言,所以有“#”号的代码都不执行。其中torch分为CPU版本和GPU版本,因为我有好一点的显卡,可以安装GPU版本,所以就在torch及torchvision前加个#号。

图3 Yolov5需要的依赖项          

3、安装Pytorch

       注意:按照此方法,Pytorch将会被安装在整个环境中,可能对其他项目的环境产生干扰。若有需要,可使用Anaconda或者Pycharm创建虚拟环境!!!创建虚拟环境后,再按照下面的步骤进行环境的配置。

Pytorch网址:Start Locally | PyTorch

       打开网页后显示的是如图4所示的界面,因为我们在“目标检测第3步:如何在Windows 10系统下安装CUDA”中安装的是CUDA11.1的版本,所以我们需要安装与CUDA版本对应的Pytorch(GPU版)。点击图4中红框超链接,可以转跳到之前的版本下载界面,界面如图5所示。

图4 Pytorch下载界面

 图5 Pytorch之前版本的下载界面

图5的界面下方包含Pytorch之前的所有版本,我们找到CUDA11.1支持的Pytorch1.8.1(torch==1.9.0也是可以的)如图6所示。

 图6 CUDA11.1支持的Pytorch1.8.1版本安装命令

再多说一下,CUDA版本的查看方式是,打开cmd,输入以下命令:

nvcc --version

然后敲击回车,即可出现以下截图,如图7所示。

图7 查看CUDA版本

将图7中复制的Pytorch安装命令复制后,打开cmd命令行,进行下载,如图8所示。(下载可能很慢,因为是国外的网站,此时可以按照附录一的方法下载。如果还是不行,可以联系我,我会热情的帮你下载!)无论是使用哪种方法下载,必须执行这一步。速度快,则继续。慢,则使用附录一的方法。

 图8 Pytorch1.8.1成功安装

 图9 安装进行中

安装成功如图10所示:

图10 pytorch安装成功

注意: 图9、图10是之前尝试CUDA11.8+Pytorch1.11.0成功安装时的截图,放在这里是供大家参考,判断什么时候才是安装Pytorch成功。这个搭配后来在YOLOv5 5.0推理时出错了,所以才更新博客,降低Pytorch的版本。

        为了进一步确保torch安装成功,我们需要在Python中验证一下。在cmd命令中输入“python”,出现>>>符号后,输入import torch,再次出现>>>符号时,则表明安装成功。除此以外,还可以键入“torch.__version__”查看信息。如图11所示:


python                           # 运行Python
import torch                     # 导入torch
torch.__version__                # 查看torch版本信息
torch.cuda.current_device()      # 查看有几个显卡设备
torch.cuda.is_available()        # 查看显卡是否激活
torch.cuda.get_device_name()     # 查看显卡设备名称

图11 验证torch安装成功操作

还可以输入“torch.cuda.current_device()”,查看CUDA,如图12所示,出现0代表只有一个显卡: 

图12 查看当前设备命令 

查看显卡名称命令行:“torch.cuda.get_device_name()”,如图13所示。 

图13 查看显卡名称 

查看显卡是否是激活状态:“torch.cuda.is_available()”,如图14所示。

图14 显卡是否激活查看命令

        至此,GPU版本的Pytorch已经安装完成。细心的同学可以发现,安装Pytorch时,torchvision也已经安装在里面了,而且也满足requirements.txt的要求。所以,可以在requirements.txt里将torchvision注释掉(加#号)。如果当时安装Pytorch选择的是附录一的方法,那么此时也得选择附录二的方法下载和安装torchversion(详见附录二)              

4、安装其他依赖项

打开requirements.txt文件,在torch和torchvision前加#号,如图15-1所示:

图15-1 备注掉torch和torchvision

复制第一行命令(不要复制#号),即

pip install -r requirements.txt

随后在对应的文件夹导航栏输入cmd,敲击回车,如图15-17所示:

 图15-2 安装其他依赖操作示意图(1)

图16 安装其他依赖操作示意图(2) 

图17 安装其他依赖操作示意图(3) 

5、下载权重

网址:Release v5.0 - YOLOv5-P6 1280 models, AWS, Supervise.ly and YouTube integrations · ultralytics/yolov5 · GitHub

打开网址后下,如图18所示。一直往下拉,找到图19所示的地方,直接点击红框圈中的文件名下载对应文件,一共有4个。将这4个权重剪切到如图20所示的目录中。

图18 首页

图19 权重截图 

 图20 目录结构 

六、运行Yolov5

       打开存有“Yolov5_master”的文件夹,如图21所示。随后,在红框位置即目录导航栏输入“cmd”,并敲击回车,输入“jupyter lab”,回车。即可显示如图22-26所示的界面。接下来,按照图的操作进行。

图21 目录结构2 

图22 jupyter lab界面1 

 图23 jupyter lab界面2

 图24 运行结果1

图25 修改权重文件路径命令

图26 代码成功运行截图 

       可以将自己的图片等资源放进“source/data/images”目录中。在运行代码时,指明文件名(包含文件后缀),重新运行代码,即可进行识别,最后会有信息提示识别的产物存储在哪里。至此Yolov5环境已安装成功。 

        细心的同学可能又发现了,解压完Yolov5的压缩包后,直接打开jupyter lab运行inference不就可以直接安装依赖了吗?是的,这是在很理想的网络条件下才可以,而且这样做安装的Pytorch版本很可能是CPU版本的。

图26-1 检验图片、视频等命令格式 

图26-1告诉我们检验摄像头、图片、视频等的方法,可以尝试一下。     

附录一 下载安装Pytorch

1.复制命令行里的网址,网址的位置如图25所示,并粘贴到浏览器的网址导航栏里。例如:

https://download.pytorch.org/whl/cu111/torch-1.8.1%2Bcu111-cp39-cp39-win_amd64.whl,

即可开始下载对应的文件。

图27 网址位置示意图 

2.将对应文件剪切到某一目录下,在该目录的导航栏处输入cmd,敲击回车,即可在对应目录

下打开cmd命令框。将文件名连同文件后缀复制,在cmd命令框中输入:“pip install xxxxx”(xxxxx代表刚才复制的内容),回

车,即可开始安装。如图28所示:

图28 安装示意图 

安装成功即可出现以下标志如图29所示,检验的命令和上面说的一样,这里的截图是初代版本的,仅供参考,不一定要完全一样。

图29  成功安装 

附录二 下载安装torchversion

1.复制命令行里的网址,网址的位置如图30所示。例如:https://download.pytorch.org/whl/cu111/torchvision-0.9.1%2Bcu111-cp39-cp39-win_amd64.whl,即可开始下载对应的文件。

图30 网址位置示意图 

2.将对应文件剪切到某一目录下,在该目录的导航栏处输入cmd,敲击回车,即可在对应目录下打开cmd命令框。将文件名连同文件后缀复制,在cmd命令框中输入:“pip install xxxxx”(xxxxx代表刚才复制的内容),回车,即可开始安装。和附录一的过程类似。就不再多叙述了。torchaudino在requirements.txt没有被要求,所以就不必理会了。 

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/531065.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

如何将Word与Endnote再次相互关联?

(请先看置顶博文)本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客 Endnote是每一个科研人员都接触…

十三、Python第十三课——类(包含对类的相关概念的理解,这一文的附录让你彻底明白类和对象)

(请先看置顶博文)https://blog.csdn.net/GenuineMonster/article/details/104495419 目录 (请先看置顶博文)https://blog.csdn.net/GenuineMonster/article/details/104495419 一、类和对象 1、类 2、对象 二、类的创建和实例…

十四、Python第十四课——文件和异常

(请先看这篇文章:https://blog.csdn.net/GenuineMonster/article/details/104495419) 如果看完这篇博文,你的问题还是没有解决,那么请关注我的公众号,后台发消息给我吧,当天回复&#x…

目标检测第6步:YOLOv5(5.0)如何在Colab中训练自定义数据集?(更新时间:2022.3.22)

(请先看这篇文章:本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客) 如果看完这…

十五、Python第十五课——测试代码

(请先看这篇文章:https://blog.csdn.net/GenuineMonster/article/details/104495419) 也许你听过软件测试?编写函数或类时,可以为其编写对应的测试代码。通过测试,可检验代码是否能够按照编写意图工作。在…

Windows 10环境下如何安装CMake?(更新时间:2021.9.13)

(请先看这篇文章:本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客) 1、下载Cmake 下…

Windows 10环境下AndroidStudio安装教程(内含如何配置Http Proxy)

(请先看这篇文章:本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客) 一、AndroidStudi…

如何将Android Studio卸载的干干净净?

(请先看这篇文章:本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客) 1、打开控制面板&…

Android Studio编译后运行按钮还是灰色的解决方法

(请先看置顶博文)本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客 安装配置好AndroidStudio之后&am…

OpenCV安装教程

(请先看置顶博文)本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客 1、下载 下载地址&#xff1a…

WordExcel使用小技巧(1)—— 打对勾或打叉

(请先看这篇文章: ​​​​​​​ 本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-C…

如何在屏幕实时显示自己键盘的输入字符?

请先看这篇文章: 本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客 大家在观看…

ERDAS IMAGINE 2015安装步骤(绝对正确,不正确,我给你远程安装)

请先看这篇文章: 本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客 1、前言 很久很久…

如何将多个PDF合成为一个PDF,并且大小统一?

一、引言 前几日,老板的申报书pdf文件及专利证明pdf文件需要合并,我使用Adobe Acrobat软件合并后发现pdf文件的页面大小会因为图片的质量不一样,具体问题是:专利证明pdf文件里的图片貌似分辨率很高,无法和申报书pdf文件…

如何设置Pycharm在打开Terminal终端时,自动进入虚拟环境?

最近旧版的Pycharm不太好用,卸载了更换了新版,但是在使用Terminal时发现,居然不是虚拟环境,我明明在设置中为项目创建了新的虚拟环境,这个问题如何解决呢? 出现这个问题的原因无非是我们的Pycharm默认了Win…

目标检测第4步:显卡、GPU、CUDA、cuDNN的介绍及如何在Windows 10下安装cuDNN?

(请先看置顶博文)本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客 目录 (请先看置顶博…

两个软件,让你对主机内显卡的信息和性能有一个更深刻的认识

(请先看置顶博文)本博打开方式!!!请详读!!!请详读!!!请详读!!!_Cat-CSDN博客 最近一段时间,显卡的…

YOLOv5-6.1识别中的BUG

1、引言 时间飞快,技术猛进,YOLOv5从2020年诞生之后,到2022年已经发展到了6.1的版本。YOLOv6和YOLOv7都已经开源,不知道小伙伴们是否有尝试? 2、问题描述及解决方法 我最近在使用YOLOv5-6.1版本时,遇到了两…

Pycharm更换清华、阿里、豆瓣软件源提高依赖包下载速度

很多同学都遇到过类似问题:用Pycharm下载某些依赖包时,速度比某度的云盘速度还低,而且还不稳定。我基本上遇不到这样的情况,但是为了日后方便,今天就记录一下更换软件源的两种方式。 一、GUI手动添加软件源 先附上常…

Windows 10系统下如何搭建轻量级目标检测NanoDet-plus环境?

NanoDet与YOLO一样,都是目标检测算法中的经典之作,在NanoDet-plus发布之后,很多CVer都争先恐后地进行尝试,感受其与NanoDet的不同。以上的截图已经将NanoDet-plus的优点完全概括,这里就不再赘述。新的plus版本精度和速…