TensorFlow2学习八之数据增强

图像增强:对图像的简单形变。

TensorFlow2图像增强函数tf.keras.preprocessing.image.ImageDataGenerator()

image_gen_train = tf.keras.preprocessing.image.ImageDataGenerator(
	rescale = 所有数据将乘以该数值
	rotation_range = 随机旋转角度数范围
	width_shift_range = 随机宽度偏移量
	height_shift_range = 随机高度偏移量
	水平翻转:horizontal_flip = 是否随机水平翻转
	随机缩放:zoom_range = 随机缩放的范围 [1-n,1+n] )
image_gen_train.fit(x_train)
x_train = x_train.reshape(x_train.shape[0], 28, 28, 1)  # 给数据增加一个维度,从(60000, 28, 28)reshape为(60000, 28, 28, 1)
image_gen_train = ImageDataGenerator(
	rescale=1. / 1., # 如为图像,分母为255时,可归至0~1
	rotation_range=45, # 随机45度旋转
	width_shift_range=.15, # 宽度偏移
	height_shift_range=.15, # 高度偏移
	horizontal_flip=False, # 水平翻转
	zoom_range=0.5 # 将图像随机缩放阈量50%)
image_gen_train.fit(x_train)

猜你喜欢

转载自blog.csdn.net/qq_41754907/article/details/113070703