Hi,
I seem to have difficulty converting ResNet to a TensorRT engine, I get the following error. Resnet should be supported since the main Tensorrt page use resnet to measure their inference improvement. Please help thanks.
Error:
Warning: No conversion function registered for layer: Merge yet.
Converting as custom op Merge bn5a_branch1/cond/Merge
name: "bn5a_branch1/cond/Merge"
op: "Merge"
input: "bn5a_branch1/cond/batchnorm/add_1"
input: "bn5a_branch1/cond/Switch_1:1"
attr {
key: "N"
value {
i: 2
}
}
attr {
key: "T"
value {
type: DT_FLOAT
}
}
Traceback (most recent call last):
File "test.py", line 81, in <module>
uff_model = uff.from_tensorflow(frozen_graph, [model_output])
File "/home/ld-sgdev/chester_liew/.virtualenvs/dl4cv/lib/python3.5/site-packages/uff/converters/tensorflow/conversion_helpers.py", line 75, in from_tensorflow
name="main")
File "/home/ld-sgdev/chester_liew/.virtualenvs/dl4cv/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py", line 64, in convert_tf2uff_graph
uff_graph, input_replacements)
File "/home/ld-sgdev/chester_liew/.virtualenvs/dl4cv/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py", line 51, in convert_tf2uff_node
op, name, tf_node, inputs, uff_graph, tf_nodes=tf_nodes)
File "/home/ld-sgdev/chester_liew/.virtualenvs/dl4cv/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py", line 28, in convert_layer
fields = cls.parse_tf_attrs(tf_node.attr)
File "/home/ld-sgdev/chester_liew/.virtualenvs/dl4cv/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py", line 177, in parse_tf_attrs
for key, val in attrs.items()}
File "/home/ld-sgdev/chester_liew/.virtualenvs/dl4cv/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py", line 177, in <dictcomp>
for key, val in attrs.items()}
File "/home/ld-sgdev/chester_liew/.virtualenvs/dl4cv/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py", line 172, in parse_tf_attr_value
return cls.convert_tf2uff_field(code, val)
File "/home/ld-sgdev/chester_liew/.virtualenvs/dl4cv/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py", line 146, in convert_tf2uff_field
return TensorFlowToUFFConverter.convert_tf2numpy_dtype(val)
File "/home/ld-sgdev/chester_liew/.virtualenvs/dl4cv/lib/python3.5/site-packages/uff/converters/tensorflow/converter.py", line 74, in convert_tf2numpy_dtype
return np.dtype(dt[dtype])
TypeError: list indices must be integers or slices, not AttrValue