ImportError: No module named 'object_detection'

[VincentDuf]

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]

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