Problem to use Tensorflow Object Detection API (Solved)

Autonomous Machines Jetson & Embedded Systems Jetson AGX Xavier
1

I’ve successfully installed tensorflow on my Xavier. However, when I try to use Tensorflow object detection API, I got this error:

$ python object_detection/builders/model_builder_test.py

  Traceback (most recent call last):
    File "object_detection/builders/model_builder_test.py", line 21, in <module>
    from object_detection.builders import model_builder
    File "/home/nvidia/models/research/object_detection/builders/model_builder.py", line 18, in <module>
    from object_detection.builders import anchor_generator_builder
    File "/home/nvidia/models/research/object_detection/builders/anchor_generator_builder.py", line 21, in <module>
    from object_detection.protos import anchor_generator_pb2
ImportError: cannot import name anchor_generator_pb2

[/code]

Can you provide an idea how to solve this problem? I used this link as reference to use this API: https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md

2

Hi,

Please remember to setup environment with this command first:

# From tensorflow/models/
protoc object_detection/protos/*.proto --python_out=.

Thanks.

3

I have the same error on drive PX2, Ubuntu 16, cuda 9.0, even after setting up environment with “protoc object_detection/protos/*.proto --python_out=.”

python3 object_detection/builders/model_builder_test.py

Traceback (most recent call last):
File “object_detection/builders/model_builder_test.py”, line 23, in
from object_detection.builders import model_builder
File “/usr/local/lib/python3.5/dist-packages/object_detection-0.1-py3.5.egg/object_detection/builders/model_builder.py”, line 20, in
from object_detection.builders import anchor_generator_builder
File “/usr/local/lib/python3.5/dist-packages/object_detection-0.1-py3.5.egg/object_detection/builders/anchor_generator_builder.py”, line 21, in
from object_detection.protos import anchor_generator_pb2
ImportError: cannot import name ‘anchor_generator_pb2’

Can you help to find a solution.