|
@@ -29,7 +29,7 @@ import json
|
|
|
from keras.preprocessing.image import ImageDataGenerator
|
|
|
from keras.models import Sequential
|
|
|
from keras.layers import Conv2D, MaxPooling2D, AveragePooling2D
|
|
|
-from keras.layers import Activation, Dropout, Flatten, Dense
|
|
|
+from keras.layers import Activation, Dropout, Flatten, Dense, BatchNormalization
|
|
|
from keras import backend as K
|
|
|
from keras.utils import plot_model
|
|
|
|
|
@@ -77,12 +77,30 @@ def generate_model():
|
|
|
model.add(MaxPooling2D(pool_size=(2, 2)))
|
|
|
|
|
|
model.add(Flatten())
|
|
|
- model.add(Dense(60))
|
|
|
+
|
|
|
+ model.add(Dense(140))
|
|
|
+ model.add(Activation('relu'))
|
|
|
+ model.add(BatchNormalization())
|
|
|
+ model.add(Dropout(0.3))
|
|
|
+
|
|
|
+ model.add(Dense(120))
|
|
|
+ model.add(Activation('relu'))
|
|
|
+ model.add(BatchNormalization())
|
|
|
+ model.add(Dropout(0.3))
|
|
|
+
|
|
|
+ model.add(Dense(80))
|
|
|
model.add(Activation('relu'))
|
|
|
- model.add(Dropout(0.4))
|
|
|
+ model.add(BatchNormalization())
|
|
|
+ model.add(Dropout(0.2))
|
|
|
+
|
|
|
+ model.add(Dense(40))
|
|
|
+ model.add(Activation('relu'))
|
|
|
+ model.add(BatchNormalization())
|
|
|
+ model.add(Dropout(0.2))
|
|
|
|
|
|
- model.add(Dense(30))
|
|
|
+ model.add(Dense(20))
|
|
|
model.add(Activation('relu'))
|
|
|
+ model.add(BatchNormalization())
|
|
|
model.add(Dropout(0.2))
|
|
|
|
|
|
model.add(Dense(1))
|