Description
I have been stuck on this error for the past 3 days. Looks minor but still unable to figure out. Anyone kindly helps me get past this.
I have trained my model on CIFAR10 on TensorFlow & then exported to ONNX. Do I need to play around with some dynamic shapes while exporting? Also, I have exported the whole “.pb” I haven’t frozen any “graph or ckpt”. Is fine. If freezing a graph or something is required kindly shed some light on that (with links).
Also, I have attached the netron output do let me know if it’s correct.onnx-model|75x500
Also, I am referring to “sample_dynamic_reshape.cpp”.
What are these formats for the images & how do I pass my CIFAR10 in such format? My CIFAR10 images are available in batches when downloaded in a binary file. How can I feed it in here ie in which format & how many images?
“train-images-idx3-ubyte”, “train-labels-idx1-ubyte”
Is it necessary to pass image in PGM / PPM? Aren’t there other ways to pass an image.
If yes what are the ways?
If No, then how do I convert each of my images in this format? I have my CIFAR10 images in NumPy array.
Here’s the GitHub link for my code & GitHub - yashraj02/Tensor-RT
Tensorflow- 2.2.0
onnx-1.7.0
tf2onnx-1.6.2
Using TensorRT Conatianer Image (20.06) Latest
CUDA <<11.0.167>>
CUDA <<11.0.167>>
<<TensorRT 7.1.2>>
Method : TernsorRT C++ API for inference
Which samples(from TensorRT C++ API) should be used for my task?
PS:
Also, a suggestion if anyone is reading from TensorRT team. Kindly add numbering (1,2,…) & sub-numbering [a,b,…] to the TensorRT GitHub Readme section. Its a nightmare for a beginner like me to get started. There are certain sections which are optional certain important can’t differentiate easily. It’s just a suggestion.
A clear and concise description of the bug or issue.
Environment
TensorRT Version 7.1.3:
GPU Type Tesla V100:
Nvidia Driver Version 450.:
CUDA Version 8.0:
CUDNN Version:
Operating System + Version Ubuntu 18.04:
Python Version (if applicable) 3.6:
TensorFlow Version (if applicable) 2.2:
PyTorch Version (if applicable):
Baremetal or Container (if container which image + tag):
Relevant Files
Please attach or include links to any models, data, files, or scripts necessary to reproduce your issue. (Github repo, Google Drive, Dropbox, etc.)
Steps To Reproduce
Please include:
- Exact steps/commands to build your repro
- Exact steps/commands to run your repro
- Full traceback of errors encountered