Tensorflow v2.3 Object Detection API fine tuning (3)

8. pre-trained 모델 다운로드

Object Detection zoo에서 다운받으시면 됩니다.

저는 EfficientDet D0 512x512을 다운받겠습니다.

object_detection 폴더에서 압축을 해제합니다.

9. pre-trained 모델 config 수정

object_detection/configs/tf2 경로에 있는 ssd_efficientdet_d0_512x512_coco17_tpu-8.config 파일을 복사해서 images 폴더에 붙여넣기 해줍니다.

방금 복사한 파일을 images 폴더에서 편집합니다.

1. num_classes -> 원하는 Object Detection 클래스 수
model {
  ssd {
    inplace_batchnorm_update: true
    freeze_batchnorm: false
    num_classes: 4
    ...

2. train_config 
-> 방금 받은 pertrained 모델의 체크포인트 경로를 적으시고(object_detection 폴더 기준)
-> detection으로 변경, batch_size는 컴퓨터가 좋다면 더 높여도 괜찮습니다.

train_config: {
  fine_tune_checkpoint: "efficientdet_d0_coco17_tpu-32/checkpoint/ckpt-0"
  fine_tune_checkpoint_version: V2
  fine_tune_checkpoint_type: "detection"
  batch_size: 4
  ...

3. train_config - optimizer -> learning rate를 낮춰줌

optimizer {
    momentum_optimizer: {
      learning_rate: {
        cosine_decay_learning_rate {
          learning_rate_base: 8e-3
          total_steps: 300000
          warmup_learning_rate: .0001
          warmup_steps: 2500
        }
      }

4. train_input_reader -> 전에 생성한 labelmap의 경로와 train.record 입력

train_input_reader: {
  label_map_path: "images/labelmap.pbtxt"
  tf_record_input_reader {
    input_path: "train.record"
  }
}

5. eval_input_reader -> 전에 생성한 labelmap의 경로와 test.record 입력

eval_input_reader: {
  label_map_path: "images/labelmap.pbtxt"
  shuffle: false
  num_epochs: 1
  tf_record_input_reader {
    input_path: "test.record"
  }
}

10. Object detection 폴더 추가 생성

models/research/object_detection 폴더에 object_detection 폴더를 같은 이름으로 생성합니다.
images 폴더를 제외하고 전부 복사 붙여넣기를 해서 넣어줍니다.
이렇게 object_detection 폴더에 다 넣어주면 추후에 import 문제가 생기지 않습니다.
물론 다른 방법도 있지만 편하게 하겠습니다.. ㅎㅎ
이 폴더는 import의 작은 문제 때문에 만든거라서 신경쓰지 않으셔도 됩니다.

11. training 폴더 생성

models/research/object_detection 폴더에 training 폴더를 생성해줍니다.
학습할 때의 정보를 이 폴더에서 사용할 예정입니다.

images/labelmap.pbtxt
images/ssd_efficientdet_d0_512x512_coco17_tpu-8.config
파일들을 복사해서 training 폴더에 붙여넣기 해줍니다
training/labelmap.pbtxt
training/ssd_efficientdet_d0_512x512_coco17_tpu-8.config