ImportError: No module named 'object_detection'

NVIDIA-AI-IOT / tf_trt_models

TensorFlow models accelerated with NVIDIA TensorRT

BSD 3-Clause "New" or "Revised" License

682 stars 245 forks source link

ImportError: No module named 'object_detection' #31

Open VincentDuf opened 5 years ago

VincentDuf commented 5 years ago

Hi, I'm trying to use the repository to test TF-TRT on the JetsonTX2 but when using python3 I try this: from tf_trt_models.detection import download_detection_model

I get the error ImportError: No module named 'object_detection'

What is weird is when I try the same command but in python (not python3), I have another error: ImportError: No module named google.protobuf The installation went all well, some ideas ? Thanks

Sri-vatsa commented 5 years ago

Add the path to object_detection in your terminal.

export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim

This is also outlined in Tensorflow Object detection API here