[DL] Tensorflow 모델 저장/불러오기

• Tensorflow 는, 두 개의 포맷 (SavedModel 과 HDF5) 으로 모델을 저장할 수 있습니다.

SavedModel 포맷으로 모델 저장/불러오기 (기본)

• SavedModel 포맷은 TF2.x 의 기본 파일 포맷입니다.

• 모델 저장하기

    # 모델 생성 및 훈련
    model = create_model()
    model.fit(train_images, train_labels, epochs=5)
  
    # 전체 모델을 기본 포맷으로 저장
    model.save('model')

• 모델 불러오기

    import tensorflow as tf
  
    # 모델 불러오기
    load_model = tf.keras.models.load_model('model')

HDF5 포맷으로 모델 저장/불러오기

• .h5(HDF5) 포맷도 모델 및 가중치를 모두 가지고 있는 데이터 파일입니다.

• 모델 저장하기

    # 모델 생성 및 훈련
    model = create_model()
    model.fit(train_images, train_labels, epochs=5)
  
    # 전체 모델을 HDF5 파일로 저장
    model.save('model.h5')

• 모델 불러오기

    import tensorflow as tf
  
    # 모델 불러오기
    load_model = tf.keras.models.load_model('model.h5')