Getting error while converting model from h5 to uff during segmentation using tensor rt


I have a Jetson Xavier NX. I am trying to run instan ce segmentation using tensor rt . However, I am getting the rror as shown below. I am following he exact instructions as shown in the github readme of the same. I want to run instance segmentation on the coco dataset.


GPU Type: Jetson Xavier NX GPU
Nvidia Driver Version: -
CUDA Version: 10.2.89
CUDNN Version: 8.0.0
Operating System + Version: Ubuntu 18.04 LTS
Python Version (if applicable): 3.6.9
TensorFlow Version (if applicable): 1.15.4
PyTorch Version (if applicable): -
Baremetal or Container (if container which image + tag): -

Relevant Files

Steps To Reproduce

Followed the steps as mentioned in Segmentation using Tensor RT

Please include:
This is the error
Traceback (most recent call last):
File “”, line 164, in
File “”, line 113, in main
model = modellib.MaskRCNN(mode=“inference”, model_dir=LOG_DIR, config=config).keras_model
File “/usr/local/lib/python3.6/dist-packages/mask_rcnn-2.1-py3.6.egg/mrcnn/”, line 1837, in init
File “/usr/local/lib/python3.6/dist-packages/mask_rcnn-2.1-py3.6.egg/mrcnn/”, line 1901, in build
File “/usr/local/lib/python3.6/dist-packages/mask_rcnn-2.1-py3.6.egg/mrcnn/”, line 180, in resnet_graph
File “/home/nvidia/.local/lib/python3.6/site-packages/keras/engine/”, line 590, in call[0])
File “/home/nvidia/.local/lib/python3.6/site-packages/keras/layers/”, line 129, in build
raise ValueError('The channel dimension of the inputs ’
ValueError: The channel dimension of the inputs should be defined. Found None.

This looks like a Jetson issue. We recommend you to raise it to the respective platform from the below link