摘要:
断点续训,即在一次训练结束后,可以先将得到的最优训练参数保存起来,待到下次训练时,直接读取最优参数,在此基础上继续训练。 读取模型参数: 存储模型参数的文件格式为 ckpt(checkpoint)。 生成 ckpt 文件时,会同步生成索引表,所以可通过判断是否存在索引表来判断是否存在模型参数。 # 阅读全文
摘要:
数据增强可以帮助扩展数据集。对图像的增强,就是对图像的简单形变,用来应对因拍照角度不同而引起的图片变形。 数据增强函数 image_gen_train = tf.keras.preprocessing.image.ImageDataGenerator( # 调整输入特征大小,每个输入特征将乘以该参数 阅读全文
摘要:
在实际应用中,我们常常需要自制数据集,解决本领域应用,而数据通常是图片或文字,需要做格式转换,才能在训练时使用。 代码: import tensorflow as tf from PIL import Image import numpy as np import os # 训练用的输入特征和标签 阅读全文