Hi,
I’m new for TensorRT. And I’m trying to convert a Inception-V3 model (the model is attached) to uff with TensorRT 4.0. However, I got the error below:
Using output node softmax
Converting to UFF graph
Warning: No conversion function registered for layer: Concat yet.
Converting as custom op Concat mixed_10/join
name: "mixed_10/join"
op: "Concat"
input: "mixed_10/join/concat_dim"
input: "mixed_10/conv"
input: "mixed_10/tower/mixed/conv"
input: "mixed_10/tower/mixed/conv_1"
input: "mixed_10/tower_1/mixed/conv"
input: "mixed_10/tower_1/mixed/conv_1"
input: "mixed_10/tower_2/conv"
attr {
key: "N"
value {
i: 6
}
}
attr {
key: "T"
value {
type: DT_FLOAT
}
}
Traceback (most recent call last):
File "tensorrt_gen.py", line 24, in <module>
uff_model =uff.from_tensorflow_frozen_model(frozen_file="./classify_image_graph_def.pb",output_nodes=["softmax"],output_filename='tmp/',text=False)
File "/usr/lib/python2.7/site-packages/uff/converters/tensorflow/conversion_helpers.py", line 113, in from_tensorflow_frozen_model
return from_tensorflow(tf_graphdef, output_nodes, **kwargs)
File "/usr/lib/python2.7/site-packages/uff/converters/tensorflow/conversion_helpers.py", line 77, in from_tensorflow
name="main")
File "/usr/lib/python2.7/site-packages/uff/converters/tensorflow/converter.py", line 74, in convert_tf2uff_graph
uff_graph, input_replacements)
File "/usr/lib/python2.7/site-packages/uff/converters/tensorflow/converter.py", line 61, in convert_tf2uff_node
op, name, tf_node, inputs, uff_graph, tf_nodes=tf_nodes)
File "/usr/lib/python2.7/site-packages/uff/converters/tensorflow/converter.py", line 31, in convert_layer
fields = cls.parse_tf_attrs(tf_node.attr)
File "/usr/lib/python2.7/site-packages/uff/converters/tensorflow/converter.py", line 201, in parse_tf_attrs
for key, val in attrs.items()}
File "/usr/lib/python2.7/site-packages/uff/converters/tensorflow/converter.py", line 201, in <dictcomp>
for key, val in attrs.items()}
File "/usr/lib/python2.7/site-packages/uff/converters/tensorflow/converter.py", line 196, in parse_tf_attr_value
return cls.convert_tf2uff_field(code, val)
File "/usr/lib/python2.7/site-packages/uff/converters/tensorflow/converter.py", line 170, in convert_tf2uff_field
return TensorFlowToUFFConverter.convert_tf2numpy_dtype(val)
File "/usr/lib/python2.7/site-packages/uff/converters/tensorflow/converter.py", line 87, in convert_tf2numpy_dtype
return np.dtype(dt[dtype])
TypeError: list indices must be integers, not AttrValue
Does it mean the ‘Concat’ layer is not supported by TensorRT 4.0? If so, the TensorRT is a little unpractical.
Any help would be appreciated!