栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

python yolov5 脚本制作(第一部分:环境搭建、yolov5源码、权重文件获取、pycharm配置、pytorch下载、初次运行yolov5代码)

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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脚本制作第一部分就算完成了,后续还会继续更新后面几部分。
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/934058.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号