TensorFlow Lite Model Maker를 사용한 이미지 분류 모델 만들기

Google COLAB 을 사용하여서 실습을 합니다. 먼저 아래 URL 에 접속을 합니다.

colab.research.google.com/github/tensorflow/tensorflow/blob/master/tensorflow/lite/g3doc/tutorials/model_maker_image_classification.ipynb?hl=ko#scrollTo=Gx1HGRoFQ54j

model_maker_image_classification.ipynb

 

첫 번째 라인에서 tflite 용 모델 패키지를 설치합니다.

 

필요한 라이브러리를 임포트 합니다.

 

Simeple End-to-End Example 를 실행합니다.

두 번째를 실행하면 이미지 분류용 주피터노트북을 확인 할 수 있습니다.

 

Run the Example 실행

ML 앱과 관련된 입력 데이터를 로드하고 훈련데이터와 테스트 데이터를 분리합니다. 2 단계에서는 TensorFlow 모델에 훈련데이터를 추가합니다.

모델에 대한 훈련과 검증이 끝나면 아래와 같은 화면이 나옵니다.

TensorFlow Lite 모델로 Export 합니다. 해당 단계에서는 모델 설명에 대한 메타 데이터를 포함한 TensorFlow Lite 모델을 생성합니다. 레이블 파일은 데이터에 포함됩니다.

model.export() 메소드를 통해 학습한 모델을 export_dir(내보내기 디렉토리)에 생성할 수 있습니다.