Skip to content

Instantly share code, notes, and snippets.

@RyanKor
Created September 24, 2021 10:06
Show Gist options
  • Select an option

  • Save RyanKor/9289a046bb36d2a2e302fc24ca29cc2c to your computer and use it in GitHub Desktop.

Select an option

Save RyanKor/9289a046bb36d2a2e302fc24ca29cc2c to your computer and use it in GitHub Desktop.
## CNN 다중 분류 이미지 sample test
import keras_preprocessing
from keras_preprocessing import image
from keras_preprocessing.image import ImageDataGenerator
training_datagen = ImageDataGenerator(
rescale = 1./255,
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')
validation_datagen = ImageDataGenerator(rescale = 1./255)
train_generator = training_datagen.flow_from_directory(
train_dir,
target_size=(227,227),
class_mode='categorical',
batch_size=126
)
validation_generator = validation_datagen.flow_from_directory(
'./tmp/test/test',
target_size=(227,227),
class_mode='categorical',
batch_size=126
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment