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

【YOLO】在ubuntu上部署yolov5

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

【YOLO】在ubuntu上部署yolov5

表格是博主使用版本:

环境版本
操作系统ubuntu 20.04 TLS
python3.8.10
pytroch1.11.0+cpu
yoloyolov5 V6.1
下载

网上看了那么多参考资料,哪有官方说明书正版,因此从官网README出发。
以下是yolov5官网的REAME中最开始的截图,图中说明了安装的流程,以及Python和Pytorch版本的限制。

具体的操作流程:

  1. 查看ubuntu的python版本
    在shell中输入python3可查看版本。

  2. 安装cpu版本pytorch(我的虚拟机ubuntu上还有显卡,因此安装cpu版本)
    官方教程(适合初次使用linux的)
    我发现了官方有在ubuntu20.04安装pytorch的教程!是全英文的,但是非常详细,包括了pip3安装等等,小白推荐这个英文教程慢慢啃:https://varhowto.com/install-pytorch-ubuntu-20-04/
    注:如果使用gpu版本(有nvidia显卡),是需要安装cuda的,cpu版本不需要安装cuda。

    简略教程(适合用过linux或是安装回顾)
    可进入pytoch官网,根据提示安装,具体可以参考YOLOV5在ubuntu18.04下安装(CPU)

    图中选择的意思是:稳定版 - linux系统 - 通过pip安装 - 语言python - cpu加速。
    在shell中输入图片最下方的命令:

    pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
    

    就开始了pytorch的安装:

    这是安装完成的示意图,我根据pytorch官方的说明,warning是不影响的。
    最后验证pytorch是否安装成功:

  3. 最后以此输入官方的命令即可(默认已经安装了git和pip)

    git clone https://github.com/ultralytics/yolov5  # clone
    cd yolov5
    pip install -r requirements.txt  # install # 网速时好时快,整一个大无语
    

预测

这里的官方说明我其实并没有看懂,但大致意思上是先下载权重模型,然后进行预测。

  1. 下载模型权重
    这里采用手动的方式下载权重模型,现在官网找到pt文件,这里选择yolov5s。
    将下载好的权重模型放入yolov5文件夹中的根目录下。

  2. 进行预测
    在默认状态下yolo提供了预测的素材,因此只要在yolov5的根目录下,终端中输入:

    python3 detect.py
    

    显示运行过程:

    显示有些包没有符合版本的要求,当还是成功运行了,结果储存在yolov5文件夹中/runs/detect/exp中。
    最后是结果展示:

    优秀链接
    Ubuntu更改镜像源两种方式

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/965347.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

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

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