Как добавить пользовательский модуль в Google Colab?

Question

Я хочу запустить обучение на данных, но оно не работает должным образом и показывает ошибку модуля:

"File "train.py", line 49, in <module>
    from object_detection.builders import dataset_builder
ModuleNotFoundError: No module named 'object_detection'"

Насколько я понимаю, пользовательские модули внутри папки обнаружения объектов не могут быть запущеныиз-за этой ошибки.Это случилось раньше.Я как-то исправил, но не помню как.Теперь я попытался добавить pythonpath или изменить рабочий каталог, но это не сработало.Может кто-нибудь помочь мне, как это исправить снова?

Answer 1

Клонируйте модели тензорного потока (только если вы этого еще не сделали)

%%bash
git clone https://github.com/tensorflow/models.git

Затем выполните следующие команды из каталога / tenorflow / models / research:

!python setup.py build   #builds the tensorflow models (this might take a while)
!python setup.py install #Install the models 
!protoc object_detection/protos/*.proto --python_out=. #protobuf compilation
cd slim
!pip install -e . #install slim
cd .. #back to research direcotry
export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim  #set the path
!python object_detection/builders/model_builder_test.py #Test the object detection model(optional)

Если все пойдет хорошо, вы увидите следующий вывод

Выполнено 22 теста за 0,152 с

ОК (пропущено = 1)

Тогда вы должны быть в состоянии запустить тренировкуСценарий без этой ошибки.

Примечание: на всякий случай, если вы столкнетесь с ошибкой, связанной с компилятором protobuf, загрузите версию 3.0.0 в каталог tenorflow / models / research и затем выполните вышеупомянутые шаги.

wget -O protobuf.zip https://github.com/google/protobuf/releases/download/v3.0.0/protoc-3.0.0-linux-x86_64.zip
unzip protobuf.zip