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')