分享好友 最新资讯首页 最新资讯分类 切换频道
jetson nano(B01)配置pytorch和torchvision环境+tensorrtx模型转换+Deepstream部署yolov5(亲测可用)
2024-12-25 17:24

jetson nano 配置pytorch和torchvision环境+tensorrt模型转换+Deepstream部署yolov5(亲测可用

jetson nano(B01)配置pytorch和torchvision环境+tensorrtx模型转换+Deepstream部署yolov5(亲测可用)

因为一次比赛接触到了jetson nano,需要使用pycharm训练自己的模型加上yolov5进行目标检测,并部署到jetson nano上,直到比赛结束也没有搞出来,后来jetson nano开始吃灰,后来因为大创需要重新开始了我的yolov5部署之路。网上资料断断续续,不是太清晰,也有太多的坑和bug,在环境配置中跌跌撞撞好几天,最后实现了目标检测,因为踩了太多的坑,于是决定写下这篇博客记录我与jetson nano和yolov5的相爱相杀。
此篇文章并不是我的原创,是我结合csdn上许多优秀博主的github上的一些优秀资源写下来的。链接我放在了最后,再次感谢这些博主给予的帮助。
以下是本篇文章正文内容,希望可以帮助大家顺利打开Yolov5的大门。

jetson nano B01 4G
USB摄像头
电脑显示屏

jetson nano下
Jetpack 4.5.1
Deepstream 5.1
windows

接下来我们进入正文
**

在此附上以上两个软件的百度网盘链接
win32Disk:
链接:https://pan.baidu.com/s/1uG8AnHu4XgOqTLLVulEhpg
提取码:gknk
SDFormatte:
链接:https://pan.baidu.com/s/1irK8jni9cE6E0meXJv_VYg
提取码:rw44
烧录完成后连接显示屏,进行一些基础的配置,配置完成后,开机。
**

本人踩了很多坑,Deepstream官网下载非常慢,需要翻墙下载。在此我也附上百度网盘链接

下载又是一个非常缓慢的过程…
我们先配置安装Deepstream所需要的环境
1.安装环境

 

2.输入以下命令以提取并安装DeepStream SDK

 

提取时一定要使用如上代码,不要将opt文件放在home目录下(后续许多环境会出错,本人亲自踩坑
**

**
1.安装pytorch1.8.0和torchvision0.9.0(切记不要随意安装版本,版本一定要对应
安装pytorch

 

安装torchvision0.9.0

 

**

**
1.克隆yolov5-4.0文件和tensorrtx文件(我这里使用的是yolov5-4.0,其他版本转换模型时可能会出错

 

2.下载最新的yolov5s.pt到yolov5目录下

 

3.将tensorrtx文件下的gen_wts.py文件复制到yolov5文件夹里

 

4.生成yolov5s.wts文件

 

执行完成后在yolov5文件夹下将会有yolov5s.wts文件生成
**

1.在tensorrt文件下创建build文件夹

 

2.将生成的 yolov5s.wts 文件移动到tensorrtx下的yolov5 文件夹中

 

3.转换为 tensorrt 模型(yolov5s.wts文件将在tensorrt/yolov5/build文件夹中生成

 

执行完成后在build文件夹下将会出现yolov5s.engine文件

4.创建自定义 yolo 文件夹并复制生成的文件(这个yolo文件夹自己创建的

 

如果创建不成功可以使用进入root权限创建
**

1.运行命令

 

这一步的意思是给文件夹赋予权限

2.将Deepstream-yolo-master下的external文件夹复制到我所创建的yolo文件下(可以使用winscp软件
在此附上Deepstream-yolo-master的百度网盘链接
链接:https://pan.baidu.com/s/1XfuIT33GCE3QElYg1cQu3A
提取码:ju2u
同时附上winscp下载链接
https://winscp.net/eng/index.php

3.进行编译

 

编译成功后,恭喜你完成了所有部署。
**

模型测试在自己创建的yolo文件夹下
输入

 
 
 

运行

 
 
 

现在是凌晨一点钟,我写下这篇博客是为了记录自己部署yolov5的经历与过程,希望在我之后部署yolov5进行目标检测的同学能够更加顺利,同时开启我的博客之路,以后我也会将自己学习到的一些东西写下一篇篇的博客记录下来。

最新文章
Chat中文版免费版网页版:开启智能对话新体验
学习辅导:无论是学生想要提升自己的学业成绩,还是职场人士需要自我提升,AI都能为用户提供个性化的学习建议,帮助其规划学习计
十三个AI绘画标签共享站点
之前已经发了一部分AI绘画的工具,社群里也有分享过一些手机端的软件,但是不同的标签生成的效果也不一样,所以这里专门整合了一
上 | 中国工程院院士孙凝晖给正国级、副国级讲课的万字长稿《人工智能与智能计算的发展》
点击蓝字关注我们上海市土木工程学会人工智能领域近年来正在迎来一场由生成式人工智能大模型引领的爆发式发展。2022年11月30日,
人工智能第一龙头,深耕数十载,中科院重仓9047.77万股!
年末大新闻来袭!科技界又出了重磅消息——Open AI和英伟达又要放大招了!这可不是空穴来风!历史告诉我们,OpenAI可是全球AI的
SEO优化必备:掌握SEO伪原创代写技巧,轻松提升排名
SEO(Search Engine Optimization)是指通过优化网站结构和内容,提高网站在搜索引擎中的排名,从而吸引更多的流量并提高转化率
2023年图灵科普最全书单 | 不信看完这个书单还不爱上科普!
我们为什么要做科普?中科院物理所研究员魏红祥认为科普是青少年认知学科、认知职业、认知世界最好的、最简单的途径。科普也是青
10 Most Important Functions of Yoast SEO for WordPress
If you have a WordPress website, and you know what SEO means, then chances are you already have the Yoast SEO WordPress
新手如何做好独立站SEO?-这篇文章手把手教会你,适合刚刚如何入门的小白。
提到独立站的SEO,相信刚入行的小伙伴的第一反应,SEO不就是写文章嘛。确实在内容为王的时代SEO中的内容营销是非常重要的一部分
电子快排系统是什么?
电子快排系统是一种在汽车换挡杆上配置的装置,其主要功能是接收和传输电子信号,通过电脑分析发动机实际转速,实现汽车降挡后自
jetson nano(B01)配置pytorch和torchvision环境+tensorrtx模型转换+Deepstream部署yolov5(亲测可用)
jetson nano 配置pytorch和torchvision环境+tensorrt模型转换+Deepstream部署yolov5(亲测可用) 因为一次