在英伟达 jeason nano配置yolov5

一、jetson nano默认已经安装了CUDA10.0，但是没有配置，需要你把CUDA的路径写入环境变量中。打开.bashrc 文件，执行命令：

sudo gedit ~/.bashrc

在文件的最后添加如下：

export CUDA_HOME=/usr/local/cuda-10.0
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-10.0/bin:$PATH

验证cuda安装是否成功：关闭当前命令行，并执行:

source ~/.bashrc
nvcc -V

查看CUDNN版本：

dpkg -l |grep -i cudnn

# 版本8.0之前:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

# 版本8.0之後（前提是确保此路径中有cudnn_version.h文件

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

附，CUDA官网下载地址：CUDA Toolkit 10.1 original Archive | NVIDIA Developer

cuDNN 官网下载连接：cuDNN Archive | NVIDIA Developer

二、配置Anaconda

主要是解决 jetson nano 的conda 环境安装问题，但是jetson nano使用的aarch64架构是没法使用anaconda的，虽然可以自己编译，但是conda指令是没法编译到aarch64架构上运行的。

不过有替代品，miniforge。

miniforge与miniconda的区别在于miniforge的下载通道是conda-forge

miniforge的地址：https://github.com/conda-forge/miniforge

miniforge的.sh release 下载地址：https://github.com/conda-forge/miniforge/releases

安装完成后，就和anaconda的使用方式一样。
主要参考链接：
手把手教你如何在Ubuntu下安装Miniconda_Python进阶者的博客-CSDN博客

三、安装Pytorch、torchvison等

首先查看下，各个库文件需要对应的版本

进入到conda环境后，开始安装需要的库

torch网址链接：https://download.pytorch.org/whl/torch_stable.html

本人用的是清华镜像下载：Links for torch (tsinghua.edu.cn)

在清华镜像里，可以下载到所有东西，不过需要根据版本一个一个去选，比较麻烦，但是如果是批量化生产，可以一次下载完后，做成bin文件去安装，就方便很多。也可以利用如下指令一步安装到位，如果大文件安装失败建议还是先去下载whl后再安装

pip install -r requirements.txt

不想下载也可以单独pip安装，缺什么库就安装什么库就可以，一般会比较顺利的进行下去，

最后，利用 pip list 查看所有已安装的库及版本信息

四、修改部分源代码

1.修改yolov5文件夹下的common.py文件

添加如图所示位置，将代码复制进去（有的话就不用复制了），记得在上方 import warnings


class SPPF(nn.Module):def __init__(self, c1, c2, k=5):super().__init__()c_ = c1 // 2self.cv1 = Conv(c1, c_, 1, 1)self.cv2 = Conv(c_ * 4, c2, 1, 1)self.m = nn.MaxPool2d(kernel_size=k, stride=1, padding=k // 2)def forward(self, x):x = self.cv1(x)with warnings.catch_warnings():warnings.simplefilter('ignore')y1 = self.m(x)y2 = self.m(y1)return self.cv2(torch.cat([x, y1, y2, self.m(y2)], 1))

2.修改环境下site-packages/torch/nn/modules/upsamping.py文件，改为如图所示：

五、运行程序测试

提前将yolov5s.pt文件下载好，可以选择不同版本，链接为：

[YOLOv5s]https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5s.pt)

[YOLOv5s]https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt)

如果是6.1版本不能用的话，就选择5.0的试试，本人在X86_64上用的是6.1的，在arm上用的5.0

1、打开笔记本电脑摄像头: python detect.py --source 0

2、打开海康摄像头：python detect.py --source rtsp://admin:xxxxxx@xxx.xxx.xxx.xxx:554/Streaming/Channels/101 --weights yolov5s.pt

3、打开大华摄像头：python detect.py --source 'rtsp://admin:xxxxxx@xxx.xxx.xxx.xxx:554/cam/realmonitor?channel=1&subtype=0' --weights yolov5s.pt

4、显示多路视频，需要新建一个streams.txt ，放在yolov5文件夹下即可，里面每个视频流放一行，类似于这样：

rtsp://admin:xxxxxx@xxx.xxx.xxx.xxx:554

rtsp://admin:xxxxxx@xxx.xxx.xxx.xxx:554/cam/realmonitor?channel=1&subtype=0

r然后运行 python detect.py --source streams.txt --weights yolov5s.pt 即可显示多路视频识别

在英伟达 jeason nano配置yolov5相关推荐

关于英伟达jetson nano的搭配双目摄像头跑ORB_SLAM2
1.安装系统按照商家给的资料安装,将Ubuntu18.04LTS镜像拷贝到tf卡中,插上jetson nano就可以安装了. 2.系统设置进入系统我先把系统语言设置为中文,在右上角的设置中找到系统 ...
英伟达Jeston nano＜3＞使用Python实现三种方式串口通信
英伟达Jeston nano<3>使用Python实现三种方式串口通信前言通信配置过程基本信息配置过程方案一:Python-40pinEXpansion Header 方案二:P ...
边缘深度学习设备基准评测：英伟达Jetson Nano胜出
点击我爱计算机视觉标星,更快获取CVML新技术本文经机器之心(微信公众号:almosthuman2014)授权转载,禁止二次转载. 选自tryolabs.com 作者:Juan Pablo 机器之 ...
用英伟达Jetson Nano运行PyTorch Fast.ai丨手把手教程+第三方工具
原作 Interogativ 郭一璞编译量子位报道 | 公众号 QbitAI 英伟达Jetson Nano,作为老黄家最便宜的硬件,只要99美元. 看到便宜就想捡,拿到手里怎么玩? 造机器人 ...
英伟达 Jetson Nano 新手必备：使用CSI或USB摄像头拍摄第一张照片
英伟达 Jetson Nano 新手必备:使用CSI或USB摄像头拍摄第一张照片 1. 引言 1.1 支持的摄像头 1.2 nvgstcapture 2. 在屏幕上开始捕获和预览显示 2.1 CSI摄 ...
英伟达Jetson Nano的初步了解
前言:跑对比实验需要不同平台,所以需要采购一个Jetson nano,但看了一下有许多版本,不太了解,故简单调研一下. 采购平台:Jetson Nano B01 Jetson Nano b01国产开发 ...
英伟达 Jetson Nano 新手必备：连接蓝牙音频
英伟达 Jetson Nano 新手必备:连接蓝牙音频 1. 介绍 2. 蓝牙适配器要求 3. 连接蓝牙音频 1. 介绍 Jetson Nano 2GB开发工具包没有内置蓝牙.为了将蓝牙音频设备连接到 ...
英伟达Jetson Nano安装
英伟达Jetson Nano安装简介板子说明 SD卡安装系统运行总结简介 NVIDIA® Jetson Nano™ 开发者套件是一款专为创客.学习者和开发者打造的小型 AI 计算机.学习此简 ...
树莓派4与英伟达Jetson Nano性能大比拼，谁是最佳的嵌入式“电脑”？
作者 | Chris Pietschmann 译者 | 弯月,责编 | 屠敏转载自CSDN(ID:CSDNnews) 导读:日前,Raspberry 基金会发布了开发者为之兴奋的 Raspberry ...

在英伟达 jeason nano配置yolov5

在英伟达 jeason nano配置yolov5相关推荐

最新文章

热门文章