Jetson Xavier NX 试玩 (二
人工智能推理模型
0 前言
就是有个函数库,怎么用呢,有使用docker的方法,有从源码编译的方法。我选择了从源码编译的方法。编译需要准备一下环境,首先在下载镜像阶段,我只下了OS,这次需要用到一些SDK COMPONENTS,所以需要先用 SDK Manager 将SDK COMPONENTS下到jetson里去。
1 下载 SDK COMPONENTS
打开可以跑 demo,虽然我也看不懂,不过似乎有点意思
2 编译 jetson library
1)更新软件源
sudo apt-get update
2)安装需要用的软件
sudo apt-get install git cmake libpython3-dev python3-numpy
3)克隆仓库
git clone --recursive https://github.com/dusty-nv/jetson-inference.git
recursive 表示将该仓库下的子仓库一起克隆下来
4)构建cmake
cd jetson-inference/
mkdir build
cd build/
cmake ../
然后就是等待,会有个图形化的图框交互出现让你选择你想下的模型,可以都取消,需要啥再下啥。当然,也可以选择要用的组件,比如 SSD-Mobilenet-v2,这个会使用 detectNet 检测动态物体,下载模型时因为网络问题可能会下载失败,因此我还是推荐不要在这个阶段下载模型,之后要用到某个模型,可以在本地下载下来再传给jetson。地址:https://github.com/dusty-nv/jetson-inference/releases 。
5)编译
make -j$(nproc
nproc就是可用的cpu核数,NX有6个cpu,但是有2个cpu默认是关闭状态,因此这里是4个
6)安装
sudo make install
7)共享动态链接库
安装完成某个工程后生成许多动态库,为了让这些动态链接库为系统所共享,还需运行动态链接库的管理命令
sudo ldconfig
8)验证
3 玩玩
detectNet
玩这个要先下 SSD-Mobilenet-v2.tar.gz,在这 https://github.com/dusty-nv/jetson-inference/releases/tag/model-mirror-190618
detectnet challenge_video.mp4
imageNet
cd ~/Desktop/jetson-inference/build/aarch64/bin
./imagenet-console来调用imageNet程序对输入图片检测,并将结果输出到指定位置
./imagenet-console ~/baiyug_ws/230312aPicture2.png ~/baiyug_ws/230312aPicture2_ouput.png
230312aPicture2_ouput.png如下
...
4 在程序中调用库接口
新建一个python程序,名为app.py
import jetson_inference
import jetson_utils
net = jetson_inference.detectNet("ssd-mobilenet-v2", th