search

FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎
GNU General Public License v2.0
1.25k stars 195 forks source link

yolov8 seg output有output0和output1，代码中使用output0有问题 #119

Open StanleyYake opened 2 months ago

StanleyYake commented 2 months ago

博主您好，你分享的代码我想用在自己训练的数据上。但是发现这里有问题，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

FeiYull commented 1 month ago

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

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