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

Could not compute dimensions for (Unnamed Layer* 214) [Convolution]_output, because the network is not valid. #15

Closed
hejun888 opened this issue Dec 23, 2020 · 4 comments

Comments

@hejun888
Copy link

hejun888 commented Dec 23, 2020

hi,
When I execute 'sudo ./yolov5 -s', the following error occurs
1

@marcoslucianops
Copy link
Owner

hi,

When I execute 'sudo ./yolov5 -s', the following error occurs

1

Do you using custom model?

@hejun888
Copy link
Author

I use my own training network, also yolov5s, classify two categories

@marcoslucianops
Copy link
Owner

I use my own training network, also yolov5s, classify two categories

Edit class_num in yololayer.h in tensorrtx folder

@hejun888
Copy link
Author

thank you very much

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