使用TensorRT加速模型推理,但在推理过程中类别显示的是class1,class2之类的,并没有显示类别名称,比如car,bus。请问需要怎么修改才能让他识别出的是类别名称而不是序号。
N你定义一个类别名称到序号的映射就行了