Hyrtsi
/ print_uint_inputs_onnx.py
Created
April 12, 2022 13:11
Python code to analyze onnx graph datatypes to solve error "Unsupported ONNX data type: UINT8 (2)"
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| """ | |
| What: | |
| Upon converting my .onnx model to TensorRT .engine I got this error | |
| Unsupported ONNX data type: UINT8 (2) | |
| So I created this short python function to check which inputs cause the error. | |
| After some googling I found out that uint8 is not supported in TensorRT. | |
| I plan on replacing the uint8's with something else so that I'm able to |