python yolov5 脚本制作(第一部分:环境搭建、yolov5源码、权重文件获取、pycharm配置、pytorch下载、初次运行yolov5代码)
开发前准备在这里先梳理一下整个脚本开发用到的东西:python解释器 / 3.7.4版本pycharm / 版本随意pytorch / 最好10.2版本 / 11.3版本yolov5代码文件 / 6.0版本anaconda / 版本随意(这个主要是用到他的那个命令提示行)开发前闲聊这次用yolov5进行开发的脚本主要是为了满足一下自己的学习兴趣,同时也是加强一下自己对于yolov5视觉神经网络算法
·
开发前准备
在这里先梳理一下整个脚本开发用到的东西:
- python解释器 / 3.7.4版本
- pycharm / 版本随意
- pytorch / 最好10.2版本 / 11.3版本
- yolov5代码文件 / 6.0版本
- anaconda / 版本随意
一、虚拟环境的创建
该部分会用到anaconda prompt,具体操作如下:
- 如图所示,打开anaconda prompt后默认的环境会在base环境下
所以我们需要做的就是创建一个虚拟环境,创建的命令是conda create -n aimbot python=3.7.4
(虚拟环境名称为:aimbot,python解释器版本:3.7.4)
当虚拟环境创造好以后,输入命令conda activate aimbot
,显示下图样式,代表虚拟环境创造成功并且成功进入虚拟环境了。
二、yolov5源码、权重文件获取
yolov5的源码需要去到github获取,下面是具体获取流程👇
- 首先先在搜索栏处搜索yolov5(下图)
- 点击相对应的项目后进去,下拉找到以前的版本👇
- 下拉找到6.0版本👇
- 选中相应版本后进入,下拉到底部下载源码和权重代码👇(注:.pt结尾的文件为权重文件)
至此yolov5源码和权重文件获取环节结束
三、pycharm配置
首先我们需要一个pycharm
pycharm官网:https://www.jetbrains.com/pycharm/download/#section=windows
(这里就不再具体讲怎么安装了)直接进入配置环节👇
- 基于上面的两步已经获取到了虚拟环境和yolov5源码以及权重文件
- 把解压好的yolov5源码文件放到pycharm 打开pycharm
- 进来后会得到以下画面👇
- 选择右下角 < No interpreter > 会得到以下画面👇 点击
add interpreter
(添加解释器)进入下一步
- 第一步先选择conda环境。第二步选择已经准备好的虚拟环境,如果一开始在anaconda prompt里面下载好虚拟环境的话这边会给你自动配置的。第三步,点击ok
至此pycharm配置环节结束
四、pytorch安装
- pytorch官网:
https://pytorch.org/
(外国网站,加载速度会比较慢)
- windows用户按照上图的选就可以了,最后把方框内的pip方法复制下来,粘贴到pycharm的terminal里面,如下图👇
- 下载后打开anaconda prompt,进入创建好的虚拟环境进行测试👇如果显示下图就证明pytorch安装成功
至此pytorch安装环节结束
五、初次运行yolov5代码
- 要运行yolov5需要用到
requirements.txt
这个文件里面的所有库,所以先把这个文件里的库先安装了👇
- 找到detect,添加之前下载好的权重文件和官方提供的被识别的文件
- 运行代码detect.py,得到如下结果证明识别成功👇
- 一起看一下结果叭。可以看到识别的阈值还是蛮高的。
那么至此为止,yolov5脚本制作第一部分就算完成了,后续还会继续更新后面几部分。
更多推荐
已为社区贡献3条内容
所有评论(0)