jetson tx2/nano上yolov4测试


注意自己的tensorrt版本,tensortt7和tensorrt8版本的项目代码不同,混用会报错。这里仅限tensorrt7版本的yolo-tensorrt。

tensorrt7和tensorrt8版本的yolo-tensorrt项目代码,及测试用到的yolov4-tiny.weights,均放在百度云盘中,请自取。

链接:https://pan.baidu.com/s/14kg0kwfikr93UuvyHs9xLw
提取码:zg4e

1、下载测试代码

在百度云盘中获取。

cd yolo-tensorrt/configs
2、下载weights

在百度云盘中获取yolov4-tiny.weights。

将weights复制进yolo-tensorrt/configs中,如图
在这里插入图片描述

3、修改代码

进入/yolo-tensorrt/samples文件夹中,修改sample_detector.cpp中的两处代码(用sudo vim)

  • 大概40行,将推理精度降低为int8,
config_v4_tiny.inference_precison = FP16;
  • 大概51行,将模型换为yolov4-tiny,
detector->init(config_v4_tiny);

如图
在这里插入图片描述

4、编译运行
cd yolo-tensorrt/
sudo mkdir build
cd build/
sudo cmake -D CMAKE_CUDA_COMPILER="/usr/local/cuda/bin/nvcc" ..
sudo make
sudo ./yolo-trt
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值