Fix densenet (make behavior follow docstring, and be consistent with …

…other applications)
akeshavan · Jan 20, 2018 · 1a78487 · 1a78487
1 parent 475def3
commit 1a78487
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/keras/applications/densenet.py b/keras/applications/densenet.py
@@ -28,6 +28,7 @@
 from ..layers import Conv2D
 from ..layers import Dense
 from ..layers import GlobalAveragePooling2D
+from ..layers import GlobalMaxPooling2D
 from ..layers import Input
 from ..layers import MaxPooling2D
 from ..layers import ZeroPadding2D
@@ -217,9 +218,9 @@ def DenseNet(blocks,
         x = Dense(classes, activation='softmax', name='fc1000')(x)
     else:
         if pooling == 'avg':
-            x = AveragePooling2D(7, name='avg_pool')(x)
+            x = GlobalAveragePooling2D(name='avg_pool')(x)
         elif pooling == 'max':
-            x = MaxPooling2D(7, name='max_pool')(x)
+            x = GlobalMaxPooling2D(name='max_pool')(x)
 
     # Ensure that the model takes into account
     # any potential predecessors of `input_tensor`.