当前位置: 七九推 > IT编程>脚本编程>Python > YOLO V5安装与验证

YOLO V5安装与验证

2022年06月23日 Python 我要评论
YOLOV5一共有4个版本,分别是YOLOV5s、YOLOV5m、YOLOV5l和YOLOV5x,其中5s的权重文件最小,检测速度也最快,我们就

YOLOV5一共有4个版本,分别是YOLOV5s、YOLOV5m、YOLOV5l和YOLOV5x,其中5s的权重文件最小,检测速度也最快,我们就在PC上配置V5s。配置过程需要下载和安装的内容如下:

  1. 安装conda环境,建议安装anaconda;
  2. 安装一个代码IDE;
  3. 安装pytorch,包括torch和torchvision;
  4. 下载YOLO源码和权重文件

第一步,安装anaconda环境

安装anaconda和代码编辑器比较简单,编辑器vscode、pycharm等都可以,anaconda在官网下载(https://www.anaconda.com)安装之后,就可以配置安装Python环境了,命令如下(环境名称为yolo,Python=3.8):

conda create --name yolo Python=3.7 -y

安装好环境后激活yolo工作环境,命令:

conda activate yolo

第二步,安装pytorch深度学习框架

pytorch官网可以根据本地PC的配置,然后在终端yolo环境下复制命令即可完成安装:

安装结束后,需要进入Python编辑环境,并输入如下验证:

import torch

print(torch.__version__)

如果没有报错,说明pytorch安装成功了(如图)。 

最后从GitHub(https://github.com/ultralytics/yolov5.git)下载YOLOV5的源码和权重文件,clone完毕之后,打开yolo文件夹内的requirements.txt文件,里面包含我们接下来要安装的配置环境和必要工具,如果当前电脑已经安装过其中部分软件,将对应的内容注释掉即可,执行命令:

pip install -r requirements.txt

等待几分钟,即可安装好yolo,默认会配置有yolo5s.pt权重配置文件,如果需要其他权重文件,可以在官网下载后放在项目文件中即可。

  • 测试yolo

用IED打开yolo项目中的detect.py,会看到项目的权重文件、输入源文件、GPU(cuda设备)以及每张图片中最大检测的目标数等等配置信息,后续自己训练样本的时候,可以根据项目的需求,调整配置信息,这里仅测试安装的结果,我们先维持当前信息不做改变。

执行detect.py文件,会看到输入结果如下:

这里的runs/detect/exp2文件夹就是我们检测后结果的存放目录(每次执行都会生成一个exp*文件夹),我们打开文件夹,会发现有两张图片,随便打开一张,上面已经显示了当前检测的类别(person、bus)、位置信息以及检测的置信度。

我们再测试一张照片(献出我的十年老狗),将需要检测的照片放到/data/images文件夹内,运行detect文件,效果如下。

 才0.77?尴尬了😅

吓得我赶紧用iDetection测试了一下。。。

(0)
打赏 微信扫一扫 微信扫一扫

相关文章:

  • 使用Python轻松实现绘制词云图项目(附详细源码)

    项目背景虽然现在已经有很多现成的制作词云图的工具了,但一般存在以下几个问题:问题一:工具太多,眼花缭乱,质量参差不齐,选择困难症;问题二:大多词云工具或多或少有一些限制,自定义的空…

    2022年06月23日 脚本编程
  • 解决Django cors跨域问题

    解决Django cors跨域问题

    django cors跨域问题前后端分离项目中的跨域问题 即同源策略同源策略:同源策略/sop(same origin policy)是一种约定,由 netsc... [阅读全文]
  • 合格的攻击性白帽黑客应该具备的基本素质(1)

    合格的攻击性白帽黑客应该具备的基本素质(1)

    很多朋友总是在问: 黑客都需要什么技术呢?想成为一个黑客应该具备什么基础呢?等等诸如此类的问题 今天有时间࿰... [阅读全文]
  • Python pyecharts实时画图自定义可视化经纬度热力图

    背景在业务数据统计分析中基本都会涉及到各省区的分析,数据可视化是数据分析的一把利器,这些省区的数据一般会用地图可视化出来,这样一些规律可以被一面了然发现地图有很多可视化类型,比如:…

    2022年06月23日 脚本编程
  • 我的内核学习笔记3:我的platform驱动模板文件

    我的内核学习笔记3:我的platform驱动模板文件

    前面的例子已经实现platform驱动了,可以将之前的字符设备驱动整合到这个驱动中,实际上我也做了这一步。因为没有什么技术含量&#x... [阅读全文]
  • 路由守卫的详解

    路由守卫的详解

    路由守卫总共有7个 全局路由守卫: beforeEach 前置守卫 affterEach 后置守卫 beforeResolve 解析守卫 路由的守卫 befo... [阅读全文]

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2022  七九推 保留所有权利. 粤ICP备17035492号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com