Description
I am using TensorRT engine for inference using Jetson TX2
I converted tensorflow freezed graph model.pb
file of face detector FaceBoxes
to faceboxes.onnx
file first.
And while creating the engine.plan
from faceboxes.onnx file, I am facing the following issue.
I have included the error log and required files as well.
Any help will be apprecieted. Thanks
Unsupported ONNX data type: UINT8 (2)
Traceback (most recent call last):
File “step-3-fb.py”, line 20, in
engine = eng.build_engine(onnx_path, shape= shape)
File “/home/nvidia/attendance-system/optimization/engine_fb.py”, line 18, in build_engine
network.get_input(0).shape = shape
AttributeError: ‘NoneType’ object has no attribute ‘shape’
Environment
TensorRT Version : 7.1.3.0
GPU Type : Jetson TX2
CUDA Version : 10.2.89
CUDNN Version : 8.0.0.180
Operating System + Version : jetpack 4.4.1
TensorFlow Version (if applicable) : 2.3.1+nv20.10
Relevant Files
faceboxes3.onnx (3.9 MB)
engine_fb.py (1.1 KB)
step-3-fb.py (625 Bytes)
model.pb (4.0 MB)