AI/Deep Learning
[DL] HDF5 와 SavedModel 간 변환
운호(Noah)
2022. 4. 10. 22:52
들어가기 앞서,
- HDF5 와 SavedModel 간 파일 포맷 변경은 간단합니다.
- 저장되어있는 모델을 Load 한 뒤, 다른 파일 포맷으로 Save 하면 됩니다.
HDF5 to SavedModel
import tensorflow as tf
model = tf.keras.models.load_model('model.h5')
model.save('model')
SavedModel to HDF5
import tensorflow as tf
model = tf.keras.models.load_model('model')
model.save('model.h5')