代码网
AI

图像分类——图像增强方法

小刘2023-07-12 21:36:3324

目录

常用的图像增强方法

在这里插入图片描述
在这里插入图片描述

tf.image进行图像增强

离线实现

复制代码 
import tensorflow as tf
import matplotlib.pyplot as plt
import 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
在这里插入图片描述

广告