【tensorrt】——语义分割推理结果不对

tensorrt, debug

问题

语义分割可视化出来的结果如下。所有位置的输出都为同一个数

可检查的方向

• 预处理
• 模型
• 后处理

解决方案

像这种所有位置都输出同一个值的情况，很大可能是输入数据的问题， 这里我打印出了输入数据，发现输入数据全是0。再往前检查，发现是初始化的时候，没有进入cuda的核函数（可以在核函数中通过printf进行输出）。解决方法就是: 指定对应的GPU_ARCHS参数，tx2:62，nx：72。