본문 바로가기

Deep Learning/Tensorflow

Tensorflow - ImageDataGenerator를 이용해서 데이터 증강하는 법

반응형
Image Augmentation

 

 

사용 코드 예시

TRAINING_DIR = '/tmp/cats-v-dogs/training'
train_datagen = ImageDataGenerator( rescale = 1 / 255.0, rotation_range = 30, width_shift_range = 0.2, height_shift_range = 0.2, shear_range = 0.2, zoom_range = 0.2, horizontal_flip = True)
train_generator =  train_datagen.flow_from_directory( TRAINING_DIR, target_size = (300,300), class_mode = 'binary', batch_size = 20 )

VALIDATION_DIR = '/tmp/cats-v-dogs/testing'
validation_datagen = ImageDataGenerator( rescale = 1 / 255.0 )
validation_generator = validation_datagen.flow_from_directory( VALIDATION_DIR, target_size = (300,300), class_mode = 'binary', batch_size = 20 )

 

 

반응형