diff --git a/Deep Learning/Dogs_vs_cats_CNN.py b/Deep Learning/Dogs_vs_cats_CNN.py index 73f0a7f..04240d4 100644 --- a/Deep Learning/Dogs_vs_cats_CNN.py +++ b/Deep Learning/Dogs_vs_cats_CNN.py @@ -127,6 +127,10 @@ def create_train_data(): model.add(Activation("relu")) model.add(MaxPooling2D(pool_size=(2,2))) +model.add(Conv2D(64, (3,3))) +model.add(Activation("relu")) +model.add(MaxPooling2D(pool_size=(2,2))) + model.add(Flatten()) model.add(Dense(64)) model.add(Activation("relu")) @@ -134,6 +138,8 @@ def create_train_data(): model.add(Dense(1)) model.add(Activation("sigmoid")) +model.summary() + model.compile(loss="binary_crossentropy", optimizer='adam', metrics=['accuracy'])