Tensorflow2로 Custom Object Detecter 학습하기 1 - 설치

섬네일 출처

해당 글은 Tensorflow 2 Object Detection API tutorial을 참고했습니다.

---

Anaconda 가상환경 사용하는 방법은 여기에서 확인할 수 있다. (Windows 10)

Target Software versions

우선 Tensorflow2 Object Detection API tutorial에서 사용한 버전을 보면 다음과 같다.

Anaconda virtual env 글에서 tensorflow 2.1을 사용했는데, 그 이유는 내 노트북에서 anaconda로 tensorflow-gpu가 설치되는 버전이 2.1이 최대였다.

tensorflow2 버전에 대해서는 뒷부분 Object Detection API를 설치에서 설명을 더 볼 수 있다.

Git clone TensorFlow Models repository

• object detection 폴더를 만들고 TensorFlow Models repository를 다운로드한다.

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

Protobuf 설치

모델 환경설정과 파라미터를 학습하기 위해 사용됨.

• protoc releases page에서 본인에게 맞는 최신 파일을 다운로드한다. (예: protoc-3.14.0-win64.zip)
• 압축 풀어서 원하는 디렉토리에 저장한다 (예: C:\Program Files\Google Protobuf)
• 시스템 변수에서 Path에 해당 경로를 추가해준다.
• 앞에서 다운받은 models/research/ 경로에서 다음 커맨드를 실행한다.

cd models/research/
protoc object_detection/protos/*.proto --python_out=.

• 커맨드 실행 후 변경된 변수를 사용하기 위해 반드시 터미널을 새로 열기.

COCO API 설치

pip install cython
pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI

Object Detection API 설치

• models/research/object_detection/packages/tf2/setup.py 파일을 models/research/ 로 복사
• python -m pip install .
• 오늘(2021/01/23) 기준 위의 커맨드를 실행하면 tensorflow2.4가 설치되는데, 내 노트북에서는 tensorflow import시 에러가 났다. 같은 오류를 발견하신 분은 virtual env에서 tensorflow2.2를 재설치하시길 권장한다.
  tensorflow: 2.2.0
  tensorflow-gpu: 2.2.0
  tensorflow-meta: 2.22.0
  tensorboard: 2.2.0
  absl-py: 0.11.0