图像分类——图像增强方法
小刘 2023-07-12 21:36:33 2023-07-12 0 0
目录
常用的图像增强方法

tf.image进行图像增强
离线实现
import tensorflow as tfimport matplotlib.pyplot as pltimport numpy as np
cat=plt.imread('./cat.jpg')plt.imshow(cat)
翻转和裁剪
#左右翻转cat1=tf.image.random_flip_left_right(cat)plt.imshow(cat1)
#上下翻转cat2=tf.image.random_flip_up_down(cat)plt.imshow(cat2)
#裁切cat3=tf.image.random_crop(cat,(200,200,3))plt.imshow(cat3)
颜色变换
#亮度调整cat4=tf.image.random_brightness(cat,0.5)plt.imshow(cat4)
#颜色色调cat5=tf.image.random_hue(cat,0.5)plt.imshow(cat5)
使用ImageDataGenerator(进行图像增强)
在线实现


for x,y in datagen.flow(x_train,y_train,batch_size=9): plt.figure(figsize=(8,8)) for i in range(0,9): plt.subplot(330+1+i) plt.imshow(x[i].reshape(28,28),cmap='gray') plt.title(y[i]) plt.show() break