딥러닝

딥러닝(16) ImageDataGenerator 사용하기

개발연습자1 2023. 1. 2. 18:00

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')

 

반응형