더북(TheBook)

flow_from_directory 메서드를 사용하면 폴더 구조를 그대로 가져와서 ImageDataGenerator에 실제 데이터를 채워 줍니다. 이때 사용되는 파라미터는 다음과 같습니다.

ⓐ 첫 번째 파라미터: 훈련 이미지 경로입니다.

target_size: 이미지 크기이며, 모든 이미지는 이 크기로 자동 조정됩니다.

color_mode: 이미지가 그레이스케일이면 'grayscale'을 사용하고, 색상이 있으면 'rgb'를 사용합니다.

batch_size: 배치당 generator에서 생성할 이미지 개수입니다.

seed: 이미지를 임의로 섞기 위한 랜덤한 숫자(정수)입니다.

shuffle: 이미지를 섞어서 사용하려면 shuffleTrue로 설정하고, 그렇지 않으면 False로 설정합니다.

class_mode: 예측할 클래스가 두 개뿐이라면 "binary"를 선택하고, 그렇지 않으면 "categorical"을 선택합니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.