tensorrt_demos: build ssd engine failed
[TensorRT] ERROR: Could not register plugin creator: FlattenConcat_TRT in namespace:
WARNING: To create TensorRT plugin nodes, please use the create_plugin_node function instead.
NOTE: UFF has been tested with TensorFlow 1.14.0.
WARNING: The version of TensorFlow installed on this system is not guaranteed to work with UFF.
UFF Version 0.6.5
=== Automatically deduced input nodes ===
[name: “Input”
op: “Placeholder”
attr {
key: “shape”
value {
shape {
dim {
size: 1
}
dim {
size: 3
}
dim {
size: 300
}
dim {
size: 300
}
}
}
}
]
Using output node NMS Converting to UFF graph Warning: No conversion function registered for layer: NMS_TRT yet. Converting NMS as custom op: NMS_TRT Warning: No conversion function registered for layer: FlattenConcat_TRT yet. Converting concat_box_conf as custom op: FlattenConcat_TRT Traceback (most recent call last): File “build_engine.py”, line 204, in <module> main() File “build_engine.py”, line 187, in main debug_mode=DEBUG_UFF) File “/usr/local/lib/python3.6/site-packages/uff/converters/tensorflow/conversion_helpers.py”, line 178, in from_tensorflow debug_mode=debug_mode) File “/usr/local/lib/python3.6/site-packages/uff/converters/tensorflow/converter.py”, line 94, in convert_tf2uff_graph uff_graph, input_replacements, debug_mode=debug_mode) File “/usr/local/lib/python3.6/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 “/usr/local/lib/python3.6/site-packages/uff/converters/tensorflow/converter.py”, line 47, in convert_layer return cls.registry_[op](name, tf_node, inputs, uff_graph, **kwargs) File “/usr/local/lib/python3.6/site-packages/uff/converters/tensorflow/converter_functions.py”, line 21, in convert_placeholder dtype = tf2uff.convert_tf2numpy_dtype(tf_node.attr[‘dtype’].type) File “/usr/local/lib/python3.6/site-packages/uff/converters/tensorflow/converter.py”, line 103, in convert_tf2numpy_dtype return tf.as_dtype(dtype).as_numpy_dtype File “/usr/local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py”, line 712, in as_dtype raise TypeError(“Cannot convert value %r to a TensorFlow DType.” % type_value) TypeError: Cannot convert value 0 to a TensorFlow DType.
linux 16.04 cuda10.0 cudnn7.6.5 Tensorrt 6.0.1.5 tensorflow 1.12 python 3.6
command line: python3 build_engine.py ssd_mobilenet_v2_egohands
About this issue
- Original URL
- State: closed
- Created 5 years ago
- Comments: 15 (6 by maintainers)
@Ndron Could you check and make sure you have patched “graphsurgeon/node_manipulation.py”? More specifically, you have to add the following line of code into the file:
Reference: