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.

Sign up for GitHub

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

Jump to bottom

make yolo编译时出现缺少输出 #30

Open
HaoWeiWang opened this issue Jul 25, 2024 · 2 comments
Open

make yolo编译时出现缺少输出 #30

HaoWeiWang opened this issue Jul 25, 2024 · 2 comments

Comments

@HaoWeiWang
Copy link

image
在运行对应的命令过后出现了上述的错误。
同时查看onnx结构，可以看到下面的结构，这是什么问题?
image
@Melody-Zhou
Copy link
Owner

@HaoWeiWang
tensorRT 报错的原因是因为找不到节点名为 output 的输出，因为内部是这么解析的，从日志打印来看你的输出节点名是 output0
但是有些奇怪，你提供的 ONNX 输出节点名又是 output，这似乎对应不上，你可能需要检查下你需要推理的 ONNX 是否正确的指定了。我这边尝试了最新代码和提供的 YOLOv8s 的 ONNX 模型能正确推理
Snipaste_2024-07-25_10-44-50

@HaoWeiWang
Copy link
Author

非常感谢回答，我再自己探索一下吧

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@Melody-Zhou @HaoWeiWang