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

   日期:2024-12-25     作者:368lu       评论:0    移动:http://g8akg8.riyuangf.com/mobile/news/7645.html
核心提示:jetson nano 配置pytorch和torchvision环境+tensorrt模型转换+Deepstream部署yolov5(亲测可用) 因为一次

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进行目标检测的同学能够更加顺利,同时开启我的博客之路,以后我也会将自己学习到的一些东西写下一篇篇的博客记录下来。

 
标签: 文件 文件夹 链接
特别提示:本信息由相关用户自行提供,真实性未证实,仅供参考。请谨慎采用,风险自负。

举报收藏 0打赏 0评论 0
 
更多>同类最新资讯
0相关评论

相关文章
最新文章
推荐文章
推荐图文
最新资讯
点击排行
{
网站首页  |  关于我们  |  联系方式  |  使用协议  |  隐私政策  |  版权隐私  |  网站地图  |  排名推广  |  广告服务  |  积分换礼  |  网站留言  |  RSS订阅  |  违规举报  |  鄂ICP备2020018471号