Hello,everyone.I am following the tutorial:
https://github.com/NVIDIA/TensorRT/tree/release/6.0/samples/opensource/sampleUffMaskRCNN
to perform inference on the Mask R-CNN network in TensorRT.
I followed the steps,when I implemented the step 7,namely “Convert the h5 model to the UFF model” I encountered a problem.Here are the information:
UFF Version 0.6.3
=== Automatically deduced input nodes ===
[name: "input_image"
op: "Placeholder"
attr {
key: "dtype"
value {
type: DT_FLOAT
}
}
attr {
key: "shape"
value {
shape {
dim {
size: -1
}
dim {
size: 3
}
dim {
size: 1024
}
dim {
size: 1024
}
}
}
}
]
=========================================
Using output node mrcnn_detection
Using output node mrcnn_mask/Sigmoid
Converting to UFF graph
Traceback (most recent call last):
File "mrcnn_to_trt_single.py", line 171, in <module>
main()
File "mrcnn_to_trt_single.py", line 130, in main
text=True, list_nodes=list_nodes)
File "mrcnn_to_trt_single.py", line 164, in convert_model
debug_mode = False
File "/home/zhaoyu/anaconda3/envs/test/lib/python3.7/site-packages/uff/converters/tensorflow/conversion_helpers.py", line 233, in from_tensorflow_frozen_model
return from_tensorflow(graphdef, output_nodes, preprocessor, **kwargs)
File "/home/zhaoyu/anaconda3/envs/test/lib/python3.7/site-packages/uff/converters/tensorflow/conversion_helpers.py", line 181, in from_tensorflow
debug_mode=debug_mode)
File "/home/zhaoyu/anaconda3/envs/test/lib/python3.7/site-packages/uff/converters/tensorflow/converter.py", line 94, in convert_tf2uff_graph
uff_graph, input_replacements, debug_mode=debug_mode)
File "/home/zhaoyu/anaconda3/envs/test/lib/python3.7/site-packages/uff/converters/tensorflow/converter.py", line 79, in convert_tf2uff_node
op, name, tf_node, inputs, uff_graph, tf_nodes=tf_nodes, debug_mode=debug_mode)
File "/home/zhaoyu/anaconda3/envs/test/lib/python3.7/site-packages/uff/converters/tensorflow/converter.py", line 47, in convert_layer
return cls.registry_[op](name, tf_node, inputs, uff_graph, **kwargs)
File "/home/zhaoyu/anaconda3/envs/test/lib/python3.7/site-packages/uff/converters/tensorflow/converter_functions.py", line 40, in convert_add
uff_graph.binary(inputs[0], inputs[1], 'add', name)
IndexError: list index out of range
I found it has created the temp.pb file, but it failed to convert .pb to .uff.
I don’t know how to deal with the problem,if you have the same problem or know how to solve it,I will be appreciated to your resopnse.