| 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]] |
| | [[분류:데이터 전처리]] | | [[분류:데이터 전처리]] |