IMAGE_SIZE = 1000

# step1:加載數(shù)據(jù)集
(train_images, train_labels), (val_images, val_labels) = tf.keras.datasets.mnist.load_data()

# step2:將圖像歸一化
train_images, val_images = train_images / 255.0, val_images / 255.0

# step3:設(shè)置訓(xùn)練集大小
train_images = train_images[:IMAGE_SIZE]
val_images = val_images[:IMAGE_SIZE]
train_labels = train_labels[:IMAGE_SIZE]
val_labels = val_labels[:IMAGE_SIZE]

# step4:將圖像的維度變?yōu)?IMAGE_SIZE,28,28,1)
train_images = tf.expand_dims(train_images, axis=3)
val_images = tf.expand_dims(val_images, axis=3)

# step5:將圖像的尺寸變?yōu)?32,32)
train_images = tf.image.resize(train_images, [32, 32])
val_images = tf.image.resize(val_images, [32, 32])

# step6:將數(shù)據(jù)變?yōu)榈?
train_loader = tf.data.Dataset.from_tensor_slices((train_images, train_labels)).batch(32)
val_loader = tf.data.Dataset.from_tensor_slices((val_images, val_labels)).batch(IMAGE_SIZE)

# step5:導(dǎo)入模型
model = LeNet5()

# 讓模型知道輸入數(shù)據(jù)的形式
model.build(input_shape=(1, 32, 32, 1))

# 結(jié)局Output Shape為 multiple
model.call(Input(shape=(32, 32, 1)))

# step6:編譯模型
model.compile(optimizer='adam',
              loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
              metrics=['accuracy'])

# 權(quán)重保存路徑
checkpoint_path = "./weight/cp.ckpt"

# 回調(diào)函數(shù)，用戶保存權(quán)重
save_callback = tf.keras.callbacks.ModelCheckpoint(filepath=checkpoint_path,
                                                   save_best_only=True,
                                                   save_weights_only=True,
                                                   monitor='val_loss',
                                                   verbose=0)

EPOCHS = 11

for epoch in range(1, EPOCHS):
    # 每個(gè)批次訓(xùn)練集誤差
    train_epoch_loss_avg = tf.keras.metrics.Mean()
    # 每個(gè)批次訓(xùn)練集精度
    train_epoch_accuracy = tf.keras.metrics.SparseCategoricalAccuracy()
    # 每個(gè)批次驗(yàn)證集誤差
    val_epoch_loss_avg = tf.keras.metrics.Mean()
    # 每個(gè)批次驗(yàn)證集精度
    val_epoch_accuracy = tf.keras.metrics.SparseCategoricalAccuracy()

    for x, y in train_loader:
        history = model.fit(x,
                            y,
                            validation_data=val_loader,
                            callbacks=[save_callback],
                            verbose=0)

        # 更新誤差，保留上次
        train_epoch_loss_avg.update_state(history.history['loss'][0])
        # 更新精度，保留上次
        train_epoch_accuracy.update_state(y, model(x, training=True))

        val_epoch_loss_avg.update_state(history.history['val_loss'][0])
        val_epoch_accuracy.update_state(next(iter(val_loader))[1], model(next(iter(val_loader))[0], training=True))

    # 使用.result()計(jì)算每個(gè)批次的誤差和精度結(jié)果
    print("Epoch {:d}: trainLoss: {:.3f}, trainAccuracy: {:.3%} valLoss: {:.3f}, valAccuracy: {:.3%}".format(epoch,
                                                                                                             train_epoch_loss_avg.result(),
                                                                                                             train_epoch_accuracy.result(),
                                                                                                             val_epoch_loss_avg.result(),
                                                                                                             val_epoch_accuracy.result()))

import model_sequential

(train_images, train_labels), (test_images, test_labels) = tf.keras.datasets.mnist.load_data()

# step2:將圖像歸一化
train_images, test_images = train_images / 255.0, test_images / 255.0

# step3:將圖像的維度變?yōu)?60000,28,28,1)
train_images = tf.expand_dims(train_images, axis=3)
test_images = tf.expand_dims(test_images, axis=3)

# step4:將圖像尺寸改為(60000,32,32,1)
train_images = tf.image.resize(train_images, [32, 32])
test_images = tf.image.resize(test_images, [32, 32])

# step5:導(dǎo)入模型
# history = LeNet5()
history = model_sequential.LeNet()

# 讓模型知道輸入數(shù)據(jù)的形式
history.build(input_shape=(1, 32, 32, 1))
# history(tf.zeros([1, 32, 32, 1]))

# 結(jié)局Output Shape為 multiple
history.call(Input(shape=(32, 32, 1)))
history.summary()

# step6:編譯模型
history.compile(optimizer='adam',
                loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
                metrics=['accuracy'])

# 權(quán)重保存路徑
checkpoint_path = "./weight/cp.ckpt"

# 回調(diào)函數(shù)，用戶保存權(quán)重
save_callback = tf.keras.callbacks.ModelCheckpoint(filepath=checkpoint_path,
                                                   save_best_only=True,
                                                   save_weights_only=True,
                                                   monitor='val_loss',
                                                   verbose=1)
# step7:訓(xùn)練模型
history = history.fit(train_images,
                      train_labels,
                      epochs=10,
                      batch_size=32,
                      validation_data=(test_images, test_labels),
                      callbacks=[save_callback])