Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- train_datagen = ImageDataGenerator(rescale=1./255, zoom_range=0.3, rotation_range=50,
- width_shift_range=0.2, height_shift_range=0.2, shear_range=0.2,
- horizontal_flip=True, fill_mode='nearest')
- val_datagen = ImageDataGenerator(rescale=1./255)
- 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)
- from keras.layers import Conv2D, MaxPooling2D, Flatten, Dense, Dropout, InputLayer
- from keras.models import Sequential
- from keras import optimizers
- model = Sequential()
- model.add(vgg_model)
- model.add(Dense(512, activation='relu', input_dim=input_shape))
- 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-5),
- 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