Tensorflow 를 이용한 Object Detection (human detection)

이번 포스팅에서는 tensorflow를 이용한 Object Detection 을 알아보려고 합니다. 

현 포스트는 본인의 필요에 의해 작업을 하는 동시에 포스팅도 작성함에 따라 조금은 비전문적일수 있음을 미리 알려드립니다. 

 

python 과 Tensorflow 환경에서 Object detection 하려고 하는데, 특이점은 linux환경이 아닌 windows환경에서 시도 하려 합니다. 

 

먼저 Tensowflow는 설치되어있어야합니다. 

간단하게 cmd창에서 다음의 명령어를 칩니다. 

pip install tensorflow-gpu (gpu버전)

 

그러고 난 후 

pip install --user Cython contextlib2  jupyter  matplotlib lxml pillow

라는 명령어를 쳐서 필요한 라이브러리들을 칩니다. 

 

tensorflow object detection API 를 다운로드 받아야됩니다. 

아래 링크에서 받아줍시다
https://github.com/tensorflow/models 

 

그리고 Protocbuffer를 다운로드 받아야 합니다. 

Protobuf의 경우는 object detection api를 컴파일 하기 위해서 필요합니다. 

https://github.com/protocolbuffers/protobuf/releases

위의 링크에서 다운로드 받으면 됩니다. git에서 다운로드 받고, 아래쪽에 win64로 받습니다. 

 

protoc 폴더

protoc.exe파일

압축을 풀어주면 protoc.exe파일이 bin폴더에 있을껍니다. 

인제 실행할때는 protoc.exe파일이 있는 디렉토리에서 실행하면 됩니다. 

 

인제 object detection api 파일을 컴파일을 해봅시다.

 

명령어는 다음과 같아요  protoc.exe이 있는 디렉토리/protoc.exe object_detection\protos\*.proto --python_out=.

예를 들어 protoc.exe 파일이 있는 디렉토리가 D:\project_folder\2019년\ex\프로그램\protoc\bin라면 빨간 글자 대신 해당 디렉토리 주소를 넣으면 됩니다. 

 

정확하게 실행되었다면 컴파일 성공이며 다음과 같이 해당 폴더에 py파일들이 생성된 것을 확인할 수 있다. 

object detection API 컴파일 결과

 

이후 Object Detedction API 의 Model\research 디렉토리로 가서

cmd창에서 해당 명령어를 쳐준다. 

 

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

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

이때 주의할 점은 CMD창은 관리자 모드로 들어가야한다. 아니면 permission error가 발생할 것이다. 

 

이후 설치가 잘되었는지 다음 명령어로 확인해본다. 

python Object Detection API가 설치된 경로\models\research\object_detection\builders/model_builder.py

아무 문제 없이 넘어갔으면 성공

 

다음 포스팅에서는 사람을 인식하는 테스트를 해보겠습니다.