def data_augmentation():
#輝度をあげる
os.system('bash ./initface.sh')
DA.sepia('sp')
DA.Shape(10, -1, 'spsp', '0') #学習画像データを水増し（データ拡張）を行う
mizumashi_data = ImageDataGenerator()
mizumashi_generator = mizumashi_data.flow_from_directory(
directory=root_dir,
target_size=(img_size, img_size),
batch_size=batch_size,
shuffle=True)
#テスト画像データを水増しする。
val_datagen = ImageDataGenerator()
val_gen = val_generator = val_datagen.flow_from_directory(
directory=root_dir,
target_size=(img_size, img_size),
batch_size=batch_size,
shuffle=False)
valX, valy = val_gen.next()
return (mizumashi_generator, val_generator, valX, valy)
#輝度をあげます
import MakeData as DA
#重複防止
# In[597]:
os.system('./initface.sh ')
# In[598]:
#水増し部分
if (args[1] == '0'):
DA.sepia('sp')
# DA.data_eraser(int(110/2))
elif (args[1] == '1'):
DA.CE_gray('red', 'red')
DA.Shape(10, -1, 'red', '0')
# DA.data_eraser(int(110/2))
elif (args[1] == '2'):
DA.CE_gray('green', 'green')
DA.Shape(10, -1, 'green', '0')
# DA.data_eraser(int(110/2))
elif (args[1] == '3'):
DA.CE_gray('blue', 'blue')
DA.Shape(10, -1, 'blue', '0')
# DA.data_eraser(int(110/2))
elif (args[1] == '4'):
DA.Shape(10, -1, 'sp', '0')
elif (args[1] == '5'):
DA.sepia('sp')
DA.Shape(10, -1, 'spsp', '0')
# In[599]:
