less than 1 minute read

Intro

  • 어제 이미지 모델은 완성했지만, non-crop 기준 정확도 60%, crop된 이미지 기준 70%가량의 정확도를 보였다.
  • 오늘은 더 나은 이미지 모델 생성을 위해 전이 학습을 진행해 볼 것이다.
  • 전이학습에 사용할 모델은 EfficientNet을 우선 적용하고, 천천히 다른 모델들로 확장시켜 나가기로 했다.

전이학습 - EfficientNet

  • Tensorflow.keras.applications에 EfficientNet 모델에 관한 데이터가 존재했다.

  • B0~B7까지의 모델중 한가지를 선택하여 작업할수 있었다.
    • 처음 프로젝트를 진행했던 2020년의 겨울에는 EfficientNetB7의 fine-tuning버전이 Image Classification 부문 sota였던것으로 기억하기 때문에 B7모델을 선택했다.
  • EfficientNetB7 모델의 input_shape는 600이므로, ImageDataGenerator의 생성 이미지 shape를 변경해주었다.

  • 추가한 코드는 다음과 같다.

<img src=https://1geraldine1.github.io/assets/images/Landmark/Day14/efficientmodel01.PNG />