NVIDIA Jetson部署YOLOv6训练环境

Jetson Xavier NX 部署YOLOv6训练环境


文章目录

  • Jetson Xavier NX 部署YOLOv6训练环境
  • 前言
  • 一、部署YOLOv6准备环境
  • 二、安装pip
  • 三、安装相关环境依赖
  • 四、onnxsim包安装问题
    • 1.解压cmake的源码包
    • 2.挂载cmake目录
    • 3.获取配置选项
    • 4.修改不必要的配置选项(看个人需要)
    • 方法1:(不保留Openssl选项)
    • 方法2:(保留Openssl选项)
    • 5.安装编译
    • 6.查看安装
    • 7.安装onnxsim
  • 五、创建文件夹放入官方训练模型
  • 六、测试运行环境
      • 最后测试结果
  • 总结


前言

在学校放假前借了一个Jetson NX的机器回家,正好最近闲的没事干来搭建一下yolov6,顺便给大家整理一下yolov6在Jetson平台上的搭建方法


一、部署YOLOv6准备环境

需要安装好目前NVIDIA Jetson系列最新的Ubuntu20.04系统,官方已经在今年五月份进行了SDKManager的更新,Python版本3.8.10

YOLOv6下载地址:https://github.com/meituan/YOLOv6 或者使用git命令:git clone https://github.com/meituan/YOLOv6.git

二、安装pip

sudo apt-get install python3-pip

三、安装相关环境依赖

在yolov6文件目录下有一个requirements.txt文档里面有相关的配置文件
NVIDIA Jetson部署YOLOv6训练环境_第1张图片

sudo pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

四、onnxsim包安装问题

由于jetson官方源给出的cmake版本太低导致onnxsim>=0.4.1版本安装不进去的问题,我们需要去cmake官网下载页面下载最新版本源码手动编译安装(只要版本大于3.21就行),这里给出资源地址:https://github.com/Kitware/CMake/releases/download/v3.24.0/cmake-3.24.0.tar.gz

1.解压cmake的源码包

tar -xvf cmake-3.24.0.tar.gz

2.挂载cmake目录

cd cmake-3.24.0

3.获取配置选项

./bootstrap

4.修改不必要的配置选项(看个人需要)

方法1:(不保留Openssl选项)

在这里我们把openssl的cmake编译选项去掉

vim Bootstrap.cmk/InitialCacheFlags.cmake

查找关键字符串 USE_OPENSSL ,把OPENSSL的配置选项改为OFF,保存文档继续使用命令

./bootstrap

方法2:(保留Openssl选项)

使用命令安装openssl相关依赖环境

sudo apt-get install libssl-dev

接着继续使用命令

./bootstrap

5.安装编译

sudo make -j6 && sudo make install

6.查看安装

cmake -version

只要版本是大于3.21就没有问题了,如果要是之前安装了其他的cmake低版本就用sudo apt-get purge cmake命令卸载掉就行了

7.安装onnxsim

输入安装命令

sudo pip install onnxsim -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

五、创建文件夹放入官方训练模型

在YOLOv6文件目录下面创建一个名为weights的文件夹
NVIDIA Jetson部署YOLOv6训练环境_第2张图片模型下载在https://github.com/meituan/YOLOv6官网开源项目里面可以找到

六、测试运行环境

在YOLOv6文件目录下,接着我们输入测试代码

python3 tools/infer.py

最后测试结果

NVIDIA Jetson部署YOLOv6训练环境_第3张图片
NVIDIA Jetson部署YOLOv6训练环境_第4张图片
NVIDIA Jetson部署YOLOv6训练环境_第5张图片

总结

提示:因为yolov6还在开发阶段,有些功能还不是很好用,出现此问题请下载最新版本的yolov6

AttributeError: Can't get attribute 'Detect' on yolov6.models.yolo'  from '/home/ubuntu/Downloads/YOLOv6/models/yolo.py'>

你可能感兴趣的:(Jetson,YOLO,python,开发语言,linux)