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

Running YoloV5 Onnx with multiple outputs #9

Closed
afisk opened this issue Jun 8, 2021 · 3 comments
Closed

Running YoloV5 Onnx with multiple outputs #9

afisk opened this issue Jun 8, 2021 · 3 comments

Comments

@afisk
Copy link

afisk commented Jun 8, 2021

I have a YoloV5 Onnx model with 4 outputs, however I can only use a single output at a time. If I try reading in all 3 in my YoloV4Prediction.cs class, I get 'Parameter is not valid' when I run predictionEngine.Predict( .. ); in Program.cs. Any ideas on how I can use all 3 output classes at the same time?
{ "output", new[] { 1, 25200, 8 } },
{ "445", new[] { 1, 3, 80, 80, 8 } },
{ "506", new[] { 1, 3, 40, 40, 8 } },
{ "567", new[] { 1, 3, 20, 20, 8 } },

@BobLd
Copy link
Owner

BobLd commented Jun 8, 2021

Hi,

Before going any further, can you confirm you had a look at the following:

Also, can you make sure your output layers names match the names in the code?

@afisk
Copy link
Author

afisk commented Jun 8, 2021

@BobLd YOU ARE AMAZING!!!!!!!!!!!!

the https://github.com/BobLd/YOLOv4MLNet/tree/yolo-v5-incl link made this super super easy. I changed a couple things and it worked like a charm!
Thank you Thank you Thank you. I've been looking at this stuff for a couple of days. THANK YOU SOOOOO MUCH!!!!!

@afisk afisk closed this as completed Jun 8, 2021
@BobLd
Copy link
Owner

BobLd commented Jun 8, 2021

My pleasure. Thanks to @raulsf6 too

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