Keras 图像数据增强

图像数据增强

导入相关的包

from keras.preprocessing.image import ImageDataGenerator,array_to_img,img_to_array,load_img
import numpy as np

定义图像生成对象

在这里插入图片描述

datagen = ImageDataGenerator(
     rotation_range=40,  #随机旋转度数
    width_shift_range = 0.2,  #随机水平平移
    height_shift_range=0.2,   #随机竖直平移
    rescale=1/255,   #数据归一化
    shear_range=0.2,  #随机错切变换
    zoom_range=0.2,   #随机放大
    horizontal_flip= True, #水平翻转
    fill_mode='nearest'  #填充方式
)

载入图片

#载入图片
img = load_img('image/content/baixiaona.jpg')
x = img_to_array(img)
print(x.shape)

(1000, 1334, 3)

x = np.expand_dims(x,0)
print(x.shape)

(1, 1000, 1334, 3)

图片生成

#生成20张图片
i = 0
for batch in datagen.flow(x,batch_size=1,save_to_dir='temp',save_format='jpeg'):
    i+=1
    if i==5:
        break
print('finished')
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值