forked from alt0xFF/hongkong_flowers
-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathaug_valid.py
31 lines (27 loc) · 1.04 KB
/
aug_valid.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import glob
import numpy as np
import os
from keras.preprocessing.image import ImageDataGenerator
from keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array, load_img
for my_class in glob.glob("./dataset/valid/**"):
save_to_dir = "./dataset/valid_aug/" + os.path.basename(my_class)
try:
os.mkdir(save_to_dir)
except:
pass
for filename in glob.glob(my_class + "/*.jpg"):
img = load_img(filename) # this is a PIL image
x = img_to_array(img) # this is a Numpy array with shape (3, 150, 150)
x = x.reshape((1,) + x.shape) # this is a Numpy array with shape (1, 3, 150, 150)
datagen = ImageDataGenerator(
rotation_range=20,
width_shift_range=0.2,
height_shift_range=0.2,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True)
i = 0
for batch in datagen.flow(x, batch_size=1, save_to_dir=save_to_dir, save_format='jpg'):
i += 1
if i > 5:
break