-
Notifications
You must be signed in to change notification settings - Fork 23
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
Problem with Infer function #3
Comments
need more log, maybe try print(x.shape)
print(self.y_tensor.shape) |
BTW, I assume that you meant |
x.shape is (85, 224, 224, 3) umm....I also print y.shape, which is (85,) Thanks in advance |
If I change self.y_tensor.shape[1] to 17, which is my number of classes, the program will run successfully. |
First, thanks for a great example of using TensorRT with Keras and TensorFlow! I have similar error with tftrt_resnet_example.py. With the TftrtEngine(TfEngine): |
Have the same issue. Shape is missing for TftrtEngine |
Hi Jeng,
Thanks for your code firstly, I am trying to use your trt_example.py code to optimize my own h5 model.
I change
frozen_graph = FrozenGraph(model, (img_h, img_w, 1))
to
frozen_graph = FrozenGraph(model, (img_h, img_w, 3))
due to my input is RGB image.
But I got the following error:
I am not sure what the problem is.
Any help from you is appreciate.
The text was updated successfully, but these errors were encountered: