guofeng.zhu
guofeng.zhu
> 你好，之前跑tensorrt官网上提供的https://github.com/NVIDIA/TensorRT/tree/release/8.5/quickstart/SemanticSegmentation 例子中，也使用了argmax，pth模型转onnx转tensorrt，可以直接可以使用；想问下，程序中有使用argmax plugin，为什么会使用argmax plugin，或者说使用argmax plugin的目的是什么？希望能帮解答一下，多谢 我理解是为了加速计算，每个像素点可以用ｃｕｄａ并行计算求最大索引类别
In similar cases, my output is all zeros