[Tensorflow Object Detection API] 설치하기 (for window)

 

먼저 Python이 설치되어있어야 합니다. 

설치 안하신분은 제 블로그 글 참고하세요 (중간에 customize installation말고 그냥 Install Now하시고, 밑에 Add Path 체크박스 클릭하세요.... 제가 썼지만 왜 그렇게 어렵게 써놨는지 이해가 안되네요...)

https://skkim1080.tistory.com/3

TensorFlow 설치하기 (+Python설치) for Window

저는 python 3.6.8버전이 설치 되어있습니다. 

 

 

1. tensorflow models 다운로드 

 

저는 git이 이미 설치 되어 있어서 git으로 다운로드 받았습니다. 

cmd 창에 다음과 같이 입력해주세요. 

 

> git clone https://github.com/tensorflow/models

 

 

2. 필요한 라이브러리 설치

 

tensorflow 설치: 저는 2.0 버전이 오류가 많이 생기는 것같아서 1.15버전 설치했습니다.

> pip install tensorflow

> pip install pilllow > pip install lxml > pip install jupyter > pip install matplotlib

 

 

3. protobuf 설치하기 + proto 파일 .py로 변환하기

 

https://github.com/google/protobuf/releases/download/v3.1.0/protoc-3.1.0-win32.zip

이 링크 클릭하시면 protoc 3.1.0버전 압축파일이 저장됩니다. 

압축 해제하시고 C:\protoc-3.1.0-win32 이렇게 C드라이브 바로 아래에 폴더를 옮겨주세요

 

아까 다운받은 models폴더의 구성을 보면

models>research>object_detection 이렇게 되어 있고 object_detection 폴더에 protos폴더가 있습니다. 

protos 폴더 안에 있는 proto 파일들을 .py 파이썬 파일로 변환해야합니다.

cmd에서 models/research 폴더로 이동해주세요.

 

C:\models\research> C:\protoc-3.1.0-win32\bin\protoc.exe object_detection\protos\*.proto --python_out=.

 

이렇게 명령어를 입력해주면 protos 폴더안에 파이썬 파일들이 생긴걸 확인할 수 있습니다. 

 

 

4. 환경 변수 설정하기

 

설치를 위해서는 PYTHONPATH를 models/research와 models/research/slim 으로 설정해야합니다. 

 

Window 환경에서 환경 변수를 설정, 수정하기 위해서는 파일 탐색기에서 내PC 마우스 우클릭>속성 으로 들어가서

 

고급 시스템 설정 클릭 후 환경 변수 클릭

 

 

시스템 변수 > PYTHONPATH 편집 > models/research폴더와 models/research/slim의 경로명을 입력하면 됩니다. 

 

 

5. setup 파일 설치

 

models/research 폴더에 보면 setup.py 파일이 있습니다. 

research 폴더로 이동한 뒤 다음 명령어를 실행합니다. 

 

C:\models\research> python setup.py build C:\models\research> python setup.py install

 

 

6. model_builder_test 파일 실행해서 설치 확인하기

 

object_detection>builders>model_builder_test.py 가 있습니다.

이 파일을 실행해서 제대로 설치가 되었는 지 확인해봅니다. 

 

C:\models\research\object_detection> python builders\model_builder_test.py

 

이런 결과가 나오면 잘 설치된 것입니다. 

 

 

7. object_detection_tutorial 실행해보기

 

이제 object_detection_tutorial을 실행해봅니다. 

object_detection_tutorial은 주피터에서 실행하는 파일이므로 object_detection 위치에서 주피터 노트북을 켜고 실행시키면 됩니다. 

 

C:\models\research\object_detection> jupyter notebook

 

주피터 노트북을 켜면 object_detection 폴더의 파일 목록이 뜰텐데 object_detection_tutorial 파일을 열어주세요

 

tutorial의 모든 소스를 실행해주면 끝입니다! (저는 맨 위에 있는 tensorflow 설치 부분은 실행하지 않았습니다.)

 

이렇게 모두 실행하고 나면 이런 결과를 볼 수 있습니다.