Jetson Nano - Tensorflow no module named 'object_detection'

I installed Tensorflow using the following command:

sudo pip3 install --pre --no-cache-dir --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu

as seen in this comment:
https://devtalk.nvidia.com/default/topic/1048776/jetson-nano/official-tensorflow-for-jetson-nano-/post/5367135/#5367135

My objective is to train an Object Detection model. I had done this before but it was on CPU only and it took a long time. I am using the Nano to hopefully speed up the process.

I am following this guide: https://towardsdatascience.com/creating-your-own-object-detector-ad69dda69c85

And I’m at the part where I’m creating the TFRecords for training. I am using the code in this link: https://github.com/datitran/raccoon_dataset/blob/master/generate_tfrecord.py

but when I run it, I can’t proceed because as the code hits this line:

from object_detection.utils import dataset_util

I get:

ModuleNotFoundError: No module named 'object_detection'

But I can import tensorflow just fine.

Any ideas?

Solved it.

I followed this answer:

https://github.com/datitran/raccoon_dataset/issues/58