在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
  2. 安装matplotlib,tqdm(python3 gen_wts.py yolov4.weight步骤需要)
pip3 install matplotlib
pip3 install tqdm

你可能感兴趣的:(神经网络)