반려견 Fine-tuning

Facebook AI Research Team에서 발표한
Model Detectron2 이용
Detectron Demo Result
detect
 {
 "dist":0.988,
 "class":"dog"
 }

[ 1개의 강아지 얼굴이 검출되었고, 고유한 강아지 얼굴 이미지가 생성 ]

Dog Detectron

모델 설정 : Detectron2에서는 미리 학습된 모델을 가져와서 fine-tuning을 수행
원하는 모델 아키텍처와 학습 설정을 설정.

Dog Detector dataset

dataset/
  ├── train/
  │ ├── class1/
  │ │ ├── image1.jpg
  │ │ ├── image2.jpg
  │ │ └── ...
  │ ├── class2/
  │ │ ├── image1.jpg
  │ │ ├── image2.jpg
  │ │ └── ...
  │ └── ...
  └── test/
  ├── class1/
  │ ├── image1.jpg
  │ ├── image2.jpg
  │ └── ...
  ├── class2/
  │ ├── image1.jpg
  │ ├── image2.jpg
  │ └── ...
  └── ...