在jetson上测试TensorRT yolov4遇到的问题（tensorrtx)

文 | MESeraph

00 | 序言

1. 在jetson NX上测试。

01 | 源码

1. 下载源码库tensorrtx，里面包含yolov4的TensorRT Engine生成代码。

2. 按yolov4给的操作执行即可。
   期间遇到的问题如下：操作之前请看下面内容，以确有序解决测试过程中遇到的问题。

02 | 安装依赖包

1. Pytorch
   Nvidia官方给出了相应的Arm包，请根据自己jetpack版本下载和安装。
   参考pytorch-for-jeston

如果安装torchvision遇到找不到#include <libavutil/pixfmt.h>
要安装ffmepg相关软件，请安装以下软件：

sudo apt install libavformat-dev
sudo apt install libavcodec-dev
sudo apt install libswresample-dev
sudo apt install libswscale-dev
sudo apt install libavutil-dev
sudo apt install libsdl1.2-dev

不要安装的话，则修改setup.py

if has ffmpeg 改为if False

2. 安装matplotlib，tqdm（python3 gen_wts.py yolov4.weight步骤需要）

pip3 install matplotlib
pip3 install tqdm