diff --git a/resnet50.py b/resnet50.py index e8cf3d7..4461623 100644 --- a/resnet50.py +++ b/resnet50.py @@ -208,7 +208,7 @@ def ResNet50(include_top=True, weights='imagenet', x = Conv2D(64, (7, 7), strides=(2, 2), name='conv1')(x) x = BatchNormalization(axis=bn_axis, name='bn_conv1')(x) x = Activation('relu')(x) - x = MaxPooling2D((3, 3), strides=(2, 2))(x) + x = MaxPooling2D((3, 3), strides=(2, 2), padding='same')(x) x = conv_block(x, 3, [64, 64, 256], stage=2, block='a', strides=(1, 1)) x = identity_block(x, 3, [64, 64, 256], stage=2, block='b')