TensorFlow lite 시작하기
Keras 모델을 저장하기 saved_model_dir = os.getcwd() + "/model" saved_model_path = saved_model_dir + '/mnistst_model' os.makedirs(saved_model_dir, exist_ok=True) model.save(saved_model_path) 위의 코드로 모델을 저장하면 "~/model/mnist_model" 폴더가 생기고 그 안에 asset 폴더, variables 폴더, 그리고 saved_model.pb 파일이 생성된다. 모델을 변환하기 TensorFlow 모델을 TensorFlow Lite 형식으로 변환하는 Python API가 제공된다. 저장되어있는 TensorFlow 모델을 읽어서 변환 후 저장하는 방법은 아래와..
SW development/Tensorflow
2020. 12. 20. 23:03