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

NetworkRT doesn't support 2D input and output layers #175

Open
nightduck opened this issue Dec 15, 2020 · 1 comment
Open

NetworkRT doesn't support 2D input and output layers #175

nightduck opened this issue Dec 15, 2020 · 1 comment

Comments

@nightduck
Copy link

The layers used in the dataDim_t struct is naturally 5 dimensional. NetworkRT reduces them to 3 by filling dimensions in with a 1. Eg 1x224x224x3x1. However, calling getBindingDimensions on a layer will return a struct that pads it with zeros, due to a quirk of a TensorRT call. dataDim_t has a method that will return the total number of elements in it, but if a dimension is zero, this total is always zero. So all 2D layers look empty.

I'm submitted a pull request so the dataDim_t struct can be initialized in a way to correct these zeros into ones.

@mive93
Copy link
Collaborator

mive93 commented Jan 4, 2021

Thank you, we'll look into that in the next days

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