search

titu1994 / DenseNet

DenseNet implementation in Keras
MIT License
707 stars 294 forks source link

average or max pooling in transition blocks? #43

Closed mvoelk closed 6 years ago

mvoelk commented 6 years ago

https://github.com/titu1994/DenseNet/blob/c1973213884613bcdf852c4fc67e30b342aba8b9/densenet.py#L530 https://github.com/titu1994/DenseNet/blob/c1973213884613bcdf852c4fc67e30b342aba8b9/densenet_fast.py#L56

titu1994 commented 6 years ago

Densenet uses Average Pooling for it's transition blocks.

mvoelk commented 6 years ago

@titu1994 You're right, but the docstrings in your code are a bit confusing. They say Max Pooling ;)

titu1994 commented 6 years ago

Ah interesting. You could add a PR to correct them since you caught it, or if you'd like i could patch this

mvoelk commented 6 years ago

I leave that to you...