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

ImageDataGenerator 사용법

 

• rescale : 크기 재조절
• rotation_range : 이미지 회전값
• zoom_range : 이미지 일부 확대
• shear_range : 이미지 기울기
• width_shift_range : 좌우 이동
• height_shift_range : 상하 이동
• horizontal_flip : 이미지 가로 뒤집기
• vertical_flip : 이미지 세로 뒤집기

from tensorflow.keras.preprocessing.image import ImageDataGenerator

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)