python yolov5 脚本制作(第一部分:环境搭建、yolov5源码、权重文件获取、pycharm配置、pytorch下载、初次运行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脚本制作第一部分就算完成了,后续还会继续更新后面几部分。
  • 来源:Chat blanc

    物联沃分享整理
    物联沃-IOTWORD物联网 » python yolov5 脚本制作(第一部分:环境搭建、yolov5源码、权重文件获取、pycharm配置、pytorch下载、初次运行yolov5代码)

    发表评论