ImageDataGenerator는 학습할 이미지 데이터가 부족할시 이미지를 편집하고 복사하여 데이터를 증강시켜주는 함수이다. 기본적으로 앞,뒤,옆모습은 기본이고 사진 뒤집기,회전,확대,축소도 된다. 이런식으로 이미지를 편집하고 증강하여 딥러닝의 정확도를 높인다.
train_datagen = ImageDataGenerator(
rotation_range=40,
width_shift_range=0.2,
height_shift_range=0.2,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True,
fill_mode='nearest')
반응형
'딥러닝' 카테고리의 다른 글
딥러닝(18) Transfer Learning과 Fine Tunning (0) | 2023.01.03 |
---|---|
딥러닝(17) 원본파일을 트레인/테스트 파일로 분리하여 사용하는 방법 (0) | 2023.01.03 |
딥러닝(15) 텐서플로우를 활용한 CNN (0) | 2023.01.02 |
딥러닝(14) 파이썬으로 압축파일 푸는 방법 (0) | 2023.01.02 |
딥러닝(13) pooling 이란 (0) | 2023.01.02 |