41번째 줄: |
41번째 줄: |
| </syntaxhighlight> | | </syntaxhighlight> |
| |- | | |- |
− | | | + | |이미지 |
− | | | + | |사진파일을 읽고, 컨텐츠를 RGB로 디코딩, 텐서로 변환, 픽셀 정규화를 한번에 해주는 도구가 있다. |
− | | | + | |<syntaxhighlight lang="python"> |
| + | from keras.preprocessing.image import ImageDataGenerator |
| + | |
| + | # 모든 이미지를 1/255로 스케일을 조정합니다 |
| + | train_datagen = ImageDataGenerator(rescale=1./255) |
| + | test_datagen = ImageDataGenerator(rescale=1./255) |
| + | |
| + | train_generator = train_datagen.flow_from_directory( |
| + | train_dir, # 타깃 디렉터리 |
| + | target_size=(150, 150), # 모든 이미지를 150 × 150 크기로 바꿉니다 |
| + | batch_size=20, |
| + | class_mode='binary') # binary_crossentropy 손실을 사용할 땐 이진 레이블. |
| + | |
| + | validation_generator = test_datagen.flow_from_directory( |
| + | validation_dir, |
| + | target_size=(150, 150), |
| + | batch_size=20, |
| + | class_mode='binary') |
| + | </syntaxhighlight> |
| |} | | |} |
| [[분류:Keras]] | | [[분류:Keras]] |
| [[분류:데이터 전처리]] | | [[분류:데이터 전처리]] |