[tensorflow] ModuleNotFoundError: No module named 'object_detection' 해결 방법

Tensorflow 레포지토리를 clone 받고 이것저것 설정 후 학습을 시작하기 위해 train.py 을 실행시켰을 때 내 경우 어떤 환경에서든 항상 'object_detection' 을 찾을 수 없다는 에러가 나왔다

 

train.py 를 실행하면 위 와 같이 'object detection' 모듈을 찾을 수 없다는 에러가 나온다

 

 

 

진짜로 저 모듈이 없는건가 싶어 tensorflow 폴더를 뒤져보면 버젓이 존재하는데 없다고 죽어버리니 당황스러웠다

 

처음엔 Pycharm 으로 실행 시키고 빨간 줄이 뜨는 곳 마다 찾아가서 object_detection 을 import 하는 곳에 모두 상위 디렉토리를 명시해 줬었다... 야근만 피할 수 있다면 노가다도 마다하지 않는다

 

하지만 환경을 세팅할 때 마다 매번 그럴 수도 없는 노릇이라 에러를 한번 검색해봤는데 다급할 땐 아무리 찾아도 나오지 않던 해결방법이 검색결과 맨 위에 나와있었다;;

 

 

방법은 간단하다

터미널 창을 열고 아래와 같이 환경변수 하나만 추가해주면 해결된다.

export PYTHONPATH=$PYTHONPATH:/home/user/tensorflow/models/research:/home/user/tensorflow/models/research/slim