I follow the instruction in the README file in objectDetector_SSD.
I’ve installed tensorflow-gpu and all the items as per the README. But I cannot convert the graph to UFF as per the instruction
# I change python to python3, /usr/lib/python2.7 to /usr/lib/python3.6
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
The output as follow
> 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
Traceback (most recent call last):
File "/usr/lib/python3.6/dist-packages/uff/bin/../../uff/converters/tensorflow/converter.py", line 19, in <module>
import tensorflow as tf
ModuleNotFoundError: No module named 'tensorflow'
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "/usr/lib/python3.6/dist-packages/uff/bin/convert_to_uff.py", line 18, in <module>
import uff
File "/usr/lib/python3.6/dist-packages/uff/bin/../../uff/__init__.py", line 2, in <module>
from uff.converters.tensorflow.conversion_helpers import from_tensorflow # noqa
File "/usr/lib/python3.6/dist-packages/uff/bin/../../uff/converters/tensorflow/conversion_helpers.py", line 12, in <module>
from .converter_functions import * # noqa
File "/usr/lib/python3.6/dist-packages/uff/bin/../../uff/converters/tensorflow/converter_functions.py", line 12, in <module>
from uff.converters.tensorflow.converter import TensorFlowToUFFConverter as tf2uff
File "/usr/lib/python3.6/dist-packages/uff/bin/../../uff/converters/tensorflow/converter.py", line 24, in <module>
https://www.tensorflow.org/install/""".format(err))
ImportError: ERROR: Failed to import module (No module named 'tensorflow')
Please make sure you have TensorFlow installed.
For installation instructions, see:
https://www.tensorflow.org/install/
Do I need to install tensorflow
seperately?
Please help!