I retrained a image classification model using tensorflow for poets(https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/)
The model i used is (https://github.com/aswinkumar2019/CNN-model-for-rupee-notes-except-2000rs-/retrained_graph.pb)
I converted it into uff format.The converted model is (https://github.com/aswinkumar2019/CNN-model-for-rupee-notes-except-2000rs-/model.uff)
When i try to infer the model using sample.py i get the error as shown in screenshot(https://github.com/aswinkumar2019/CNN-model-for-rupee-notes-except-2000rs-/send.png)
[TensorRT] ERROR: UffParser: Unsupported number of graph 0
[TensorRT] ERROR: Network must have at least one output
Traceback (most recent call last):
File “sample.py”, line 112, in
File “sample.py”, line 98, in main
with build_engine(model_file) as engine:
Is model retrained using tensorflow for poets not supported in tensorrt???
Please help me.