Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
StanleyYake opened this issue Apr 25, 2024 · 1 comment

Comments

@StanleyYake
Copy link

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

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

@FeiYull
Copy link
Owner

FeiYull commented May 16, 2024

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

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants