fixed average pooling to max pooling

titu1994 / Fully-Connected-DenseNets-Semantic-Segmentation

Fully Connected DenseNet for Image Segmentation (https://arxiv.org/pdf/1611.09326v1.pdf)

84 stars 41 forks source link

Closed Timonzimm closed 7 years ago

Timonzimm commented 7 years ago

Changed AveragePooling2D to MaxPooling2D referring to the official paper https://arxiv.org/pdf/1611.09326.pdf (page 5, Table 1).

titu1994 commented 7 years ago

I'm going to add a flag to state which type of pooling can be done.

I have seen a few cases where average performs better than max. Once I correct the implementation, I'll add the flag.