Chinaunix首页 | 论坛 | 博客
  • 博客访问: 3711894
  • 博文数量: 356
  • 博客积分: 10458
  • 博客等级: 上将
  • 技术积分: 4734
  • 用 户 组: 普通用户
  • 注册时间: 2008-03-24 14:59
文章分类

全部博文(356)

文章存档

2020年(17)

2019年(9)

2018年(26)

2017年(5)

2016年(11)

2015年(20)

2014年(2)

2013年(17)

2012年(15)

2011年(4)

2010年(7)

2009年(14)

2008年(209)

分类: Python/Ruby

2019-08-02 17:23:18

可批量对图片进行旋转,增大,缩小等操作,生成新的图片
假设目录如下:
├── train_gen
└── train_org
└── sheep
└── 0.jpg
train_org是原始数据,子目录sheep为标签(注意源目录下必须要有子目录做为标签,即y),0.jpg为输入数据(即train_x)。 批量生成图片到train_gen/下


python代码

点击(此处)折叠或打开

  1. from keras.preprocessing import image

  2. srcPath='/home/zm/download/jpg/test_gen/train_org/'
  3. dstPath='/home/zm/download/jpg/test_gen/train_gen/'

  4. dataGen = image.ImageDataGenerator(rescale=1. / 255, rotation_range = 45, zoom_range = 0.5)
  5. genData = dataGen.flow_from_directory(srcPath, batch_size=1, shuffle=False, save_to_dir=dstPath,
  6.                                        save_prefix='gen', target_size=(224, 224))

  7. for i in range(5):
  8.     genData.next()

执行
python3 ./img_gen.py
输出
Using TensorFlow backend.
Found 1 images belonging to 1 classes.

文件变化成如下
├── train_gen
│ ├── gen_0_1566435.png
│ ├── gen_0_2240910.png
│ ├── gen_0_4362670.png
│ ├── gen_0_6886943.png
│ └── gen_0_9432471.png
└── train_org
└── sheep
└── 0.jpg


作者:帅得不敢出门 

阅读(1895) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~