search

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

Average Pooling should be Max Pooling? #6

Closed ahundt closed 7 years ago

ahundt commented 7 years ago

It looks like the transition_down_block uses AveragePooling2D when it should be using MaxPooling2D as specified in the paper 3.3. Semantic Segmentation Architecture.

https://github.com/titu1994/Fully-Connected-DenseNets-Semantic-Segmentation/blob/master/densenet_fc.py#L75

Though perhaps that's worth making configurable with a flag.

titu1994 commented 7 years ago

Hmm true. I overlooked the pooling method in the original implementation. Could you add a PR if you are free ? Or I will update in a few days.