Google Colaboratoryを使ってTensorFlow2 Object Detection APIを使いたいが最後の動作確認でエラーになる

Question

TensorFlow2 Object Detection APIのチュートリアルを進めていくにあたり、こちらの動画を参照しています。
https://www.youtube.com/watch?v=MzG3EbT_-2Y

%cd /content/gdrive/MyDrive/TensorFlow/workspace/TensorTest/TensorFlowApi/models/research
!protoc object_detection/protos/*.proto --python_out=.
%env PYTHONPATH=/env/python:/content/gdrive/MyDrive/TensorFlow/workspace/TensorTest/TensorFlowApi/models/research/slim
%env
!python object_detection/builders/model_builder_tf2_test.py

最後の動作確認において上記のコードを入力しているのですが、

/content/gdrive/MyDrive/TensorFlow/workspace/TensorTest/TensorFlowApi/models/research
env: PYTHONPATH=/env/python:/content/gdrive/MyDrive/TensorFlow/workspace/TensorTest/TensorFlowApi/models/research/slim
Traceback (most recent call last):
  File "object_detection/builders/model_builder_tf2_test.py", line 24, in <module>
    from object_detection.builders import model_builder
  File "/usr/local/lib/python3.7/dist-packages/object_detection/builders/model_builder.py", line 23, in <module>
    from object_detection.builders import anchor_generator_builder
  File "/usr/local/lib/python3.7/dist-packages/object_detection/builders/anchor_generator_builder.py", line 26, in <module>
    from object_detection.protos import anchor_generator_pb2
ImportError: cannot import name 'anchor_generator_pb2' from 'object_detection.protos' (/usr/local/lib/python3.7/dist-packages/object_detection/protos/__init__.py)

という風に返ってき、うまく実行されません。
動画の通りに書いているつもりなのですが、何が足りないのか、何が間違っているのか分かりません。

お手数ですが、コードの修正をしていただけると幸いです。

Comments

[jbpb0]

エラーメッセージを見ると、「object_detection」というディレクトリが
/content/gdrive/MyDrive/TensorFlow/workspace/TensorTest/TensorFlowApi/models/research/object_detection
と、
/usr/local/lib/python3.7/dist-packages/object_detection
に有って、

> File "object_detection/builders/model_builder_tf2_test.py", line 24, in <module>

は前者に有るファイルのことで、

> File "/usr/local/lib/python3.7/dist-packages/object_detection/builders/model_builder.py", line 23, in <module>

と

> File "/usr/local/lib/python3.7/dist-packages/object_detection/builders/anchor_generator_builder.py", line 26, in <module>

は後者に有るファイルのことです

詳細理解してないので外してるかもしれませんが、前者に有るファイルが使われないといけないところで後者に有るファイルが使われてる、ということはないですかね

たとえば、

> File "/usr/local/lib/python3.7/dist-packages/object_detection/builders/model_builder.py", line 23, in <module>

は、上記ファイルではなく
/content/gdrive/MyDrive/TensorFlow/workspace/TensorTest/TensorFlowApi/models/research/object_detection/builders/model_builder.py
が使われないといけないのではないか、と思いまして

[jbpb0]

PYTHONPATH=/env/python:/content/gdrive/MyDrive/TensorFlow/workspace/TensorTest/TensorFlowApi/models/research/slim
に
/content/gdrive/MyDrive/TensorFlow/workspace/TensorTest/TensorFlowApi/models/research
を追加したら、どうでしょうか？

【追記】
https://medium.com/analytics-vidhya/tensorflow-2-object-detection-api-using-custom-dataset-745f30278446
を見ると、
/content/gdrive/MyDrive/TensorFlow/workspace/TensorTest/TensorFlowApi/models/research/object_detection
も追加するのかも

[ttttt08]

早々の回答ありがとうございます。

２つ提示していただきましたが、２つとも動作確認することが出来ました！
また、分かりやすく解説していただき、理解を深めることが出来ました。
ありがとうございます。

Answer 1

PYTHONPATH=/env/python:/content/gdrive/MyDrive/TensorFlow/workspace/TensorTest/TensorFlowApi/models/research/slim
に
/content/gdrive/MyDrive/TensorFlow/workspace/TensorTest/TensorFlowApi/models/research
を追加

　
上記でダメなら、
/content/gdrive/MyDrive/TensorFlow/workspace/TensorTest/TensorFlowApi/models/research/object_detection
も追加
TensorFlow 2 Object Detection API using Custom Dataset