Following the instructions in the README (/opt/nvidia/deepstream/deepstream/sources/objectDetector_SSD/README), when I issue the command:
$ python3 /usr/lib/python3.6/dist-packages/uff/bin/convert_to_uff.py \ frozen_inference_graph.pb -O NMS \ -p /usr/src/tensorrt/samples/sampleUffSSD/config.py \ -o sample_ssd_relu6.uff
It fails with this error:
2020-08-07 14:49:01.364038: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcudart.so.10.2 Loading frozen_inference_graph.pb Traceback (most recent call last): File "/usr/lib/python3.6/dist-packages/uff/bin/convert_to_uff.py", line 143, in <module> main() File "/usr/lib/python3.6/dist-packages/uff/bin/convert_to_uff.py", line 139, in main debug_mode=args.debug File "/usr/lib/python3.6/dist-packages/uff/bin/../../uff/converters/tensorflow/conversion_helpers.py", line 274, in from_tensorflow_frozen_model with tf.gfile.GFile(frozen_file, "rb") as frozen_pb: AttributeError: module 'tensorflow' has no attribute 'gfile'
I have installed the latest tensorflow (2.2.0) as per the README.
I can fix this with an edit as follows to line #274 of
with tf.gfile.GFile(frozen_file, "rb") as frozen_pb:
with tf.io.gfile.GFile(frozen_file, "rb") as frozen_pb:
But there are further errors…
Warning: No conversion function registered for layer: NMS_TRT yet. AttributeError: module 'tensorflow' has no attribute 'AttrValue'
This used to work just fine in 5.0 developer preview so what has happened? Have the instructions not been updated or something???
Please provide complete information as applicable to your setup.
• Hardware Platform (Jetson / GPU) Jetson Nano
• DeepStream Version 5.0GA