使用new_model = tf.keras.models.load_model('my_model.h5') 创建一个与 my_model相同的模型并加载权重测试。
创建 model_hand_h5_load.py
#encoding=utf-8
# 手动创建和加载 *.h5 模型和权重值
from __future__ import absolute_import, division, print_function, unicode_literals
import os
import tensorflow as tf
from tensorflow import keras
print(tf.version.VERSION)
(train_images, train_labels), (test_images, test_labels) = tf.keras.datasets.mnist.load_data()
train_labels = train_labels[:1000]
test_labels = test_labels[:1000]
train_images = train_images[:1000].reshape(-1, 28 * 28) / 255.0
test_images = test_images[:1000].reshape(-1, 28 * 28) / 255.0
# # 定义一个简单的序列模型
# def create_model():
# model = tf.keras.models.Sequential([
# keras.layers.Dense(512, activation='relu', input_shape=(784,)),
# keras.layers.Dropout(0.2),
# keras.layers.Dense(10, activation='softmax')
# ])
# return model
#