i want to get uff file from yolo_v3 pb file, but i get some errors

AI & Data Science Deep Learning (Training & Inference) TensorRT
1 
import tensorflow as tf
import uff
uff.from_tensorflow_frozen_model(
    "/media/lw/OS/yolo_v3.pb",
    "outputboxes,outputscores,outputclasses".split(',')
)

errors

NOTE: UFF has been tested with TensorFlow 1.12.0. Other versions are not guaranteed to work
UFF Version 0.6.3
=== Automatically deduced input nodes ===
[name: “Placeholder”
op: “Placeholder”
attr {
key: “dtype”
value {
type: DT_INT32
}
}
attr {
key: “shape”
value {
shape {
dim {
size: 2
}
}
}
}
, name: “input_image”
op: “Placeholder”
attr {
key: “dtype”
value {
type: DT_FLOAT
}
}
attr {
key: “shape”
value {
shape {
dim {
size: -1
}
dim {
size: 416
}
dim {
size: 416
}
dim {
size: 3
}
}
}
}
]

Using output node outputboxes
Using output node outputscores
Using output node outputclasses
Converting to UFF graph
Warning: No conversion function registered for layer: Fill yet.
Converting ones_like_2 as custom op: Fill
Warning: No conversion function registered for layer: NonMaxSuppressionV3 yet.
Converting non_max_suppression_2/NonMaxSuppressionV3 as custom op: NonMaxSuppressionV3
Warning: No conversion function registered for layer: Where yet.
Converting boolean_mask_5/Where as custom op: Where
Warning: No conversion function registered for layer: GreaterEqual yet.
Converting GreaterEqual as custom op: GreaterEqual
Traceback (most recent call last):
File “/home/lw/PycharmProjects/lenet/get_node_op.py”, line 19, in
“outputboxes,outputscores,outputclasses”.split(‘,’)
File “/home/lw/anaconda3/lib/python3.6/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/lw/anaconda3/lib/python3.6/site-packages/uff/converters/tensorflow/conversion_helpers.py”, line 181, in from_tensorflow
debug_mode=debug_mode)
File “/home/lw/anaconda3/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 “/home/lw/anaconda3/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 “/home/lw/anaconda3/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 “/home/lw/anaconda3/lib/python3.6/site-packages/uff/converters/tensorflow/converter_functions.py”, line 580, in convert_strided_slice
raise ValueError(“ellipsis_mask not supported”)
ValueError: ellipsis_mask not supported

How should I solve this problem?
thansk!

2

Hi,

It looks like the model you are converting to TensorRT contains many unsupported layers and operations "ex: NonMaxSuppressionV3, GreaterEqual, Where, etc ".

For a list of supported operations, please reference: [url]https://docs.nvidia.com/deeplearning/sdk/tensorrt-support-matrix/index.html#supported-ops[/url]

For unsupported layers, users can extend TensorRT functionalities by implementing custom layers using the IPluginV2 class for the C++ and Python API. Custom layers, often referred to as plugins, are implemented and instantiated by an application, and their lifetime must span their use within a TensorRT engine. [url]https://docs.nvidia.com/deeplearning/sdk/tensorrt-developer-guide/index.html#extending[/url]

Thanks,
NVIDIA Enterprise Support