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'>