Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- train_generator = train_datagen.flow(train_imgs, train_labels_enc, batch_size=30)
- val_generator = val_datagen.flow(validation_imgs, validation_labels_enc, batch_size=20)
- input_shape = (150, 150, 3)
- from keras.layers import Conv2D, MaxPooling2D, Flatten, Dense, Dropout
- from keras.models import Sequential
- from keras import optimizers
- model = Sequential()
- model.add(Conv2D(16, kernel_size=(3, 3), activation='relu',
- input_shape=input_shape))
- model.add(MaxPooling2D(pool_size=(2, 2)))
- model.add(Conv2D(64, kernel_size=(3, 3), activation='relu'))
- model.add(MaxPooling2D(pool_size=(2, 2)))
- model.add(Conv2D(128, kernel_size=(3, 3), activation='relu'))
- model.add(MaxPooling2D(pool_size=(2, 2)))
- model.add(Conv2D(128, kernel_size=(3, 3), activation='relu'))
- model.add(MaxPooling2D(pool_size=(2, 2)))
- model.add(Flatten())
- model.add(Dense(512, activation='relu'))
- model.add(Dropout(0.3))
- model.add(Dense(512, activation='relu'))
- model.add(Dropout(0.3))
- model.add(Dense(1, activation='sigmoid'))
- model.compile(loss='binary_crossentropy',
- optimizer=optimizers.RMSprop(lr=1e-4),
- metrics=['accuracy'])
- history = model.fit_generator(train_generator, steps_per_epoch=100, epochs=100,
- validation_data=val_generator, validation_steps=50,
- verbose=1)
Add Comment
Please, Sign In to add comment