TensorRT-Alpha icon indicating copy to clipboard operation
TensorRT-Alpha copied to clipboard

yolov8 seg output有output0和output1,代码中使用output0有问题

Open StanleyYake opened this issue 1 year ago • 1 comments

博主您好,你分享的代码我想用在自己训练的数据上。但是发现这里有问题,onnx转换为TensorRT的trt文件成功后,我用Netron查看了数据:它的输出有两种呢, Snipaste_2024-04-25_16-02-25 可是你的代码中只用了一种,这里没有问题吗? https://github.com/FeiYull/TensorRT-Alpha/blob/bca9575229ef5f6fe4c5acf51c1bd3c7e5959ec6/yolov8-seg/app_yolov8_seg.cpp#L13

StanleyYake avatar Apr 25 '24 08:04 StanleyYake

@StanleyYake 还真是,但是不影响,可以自己手动改成:

initParameters.input_output_names = { "images", "output0", ""output1"" };

FeiYull avatar May 16 '24 02:05 FeiYull