인프런 - 강의/나도 만들어본다 AI 앱 (tensorflow+android)

9 - Tensorflow lite 만들기

개복치 개발자 2020. 3. 10. 02:08

텐서플로우 라이트란 무엇일까요?

 

 

텐서플로우로 만든 머신러닝 모델을 모바일에 할 수 있게 해놓은 프레임워크입니다.

 

열심히 저희가 만들어놓은 머신러닝 모델을 모바일에서 사용할 수 있도록 변경해서, 사용합니다.

 

얘를 어떻게 쓰느냐 하면, 아래와 같은 절차로 진행됩니다.

 

 

 

 

 

converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_model = converter.convert()
open("mnist.tflite", "wb").write(tflite_model)

 

from google.colab import files
files.download( "mnist.tflite" )