Jetson Xavier NX 部署YOLOv6训练环境

前言

在学校放假前借了一个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文档里面有相关的配置文件

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的文件夹
模型下载在https://github.com/meituan/YOLOv6官网开源项目里面可以找到

六、测试运行环境

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

python3 tools/infer.py

最后测试结果

总结

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

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