[Tensorflow & Android] .h5 -> .pb -> .tflite로 변환 & 안드로이드에 적용(python & java)
.h5 -> .pb -> .tflite로 변환 & 안드로이드에 적용(python & java) 1. .h5 -> .pb -> .tflite로 변환 .h5 파일 -> .pb 파일로 변환하기 (코드) from tensorflow import keras model = keras.models.load_model('./model/my_model.h5', compile=False) export_path = './pb' model.save(export_path, save_format="tf") model = keras.models.load_model('#######', compile=False) → ######에는 .h5파일이 있는곳의 directory를 쓴다. export_path = '@@@' → @@@에는 변환..
IT Convergence Engineering/AI 버섯 어플
2020. 7. 28. 01:32