search

zijundeng / pytorch-semantic-segmentation

PyTorch for Semantic Segmentation
MIT License
1.72k stars 395 forks source link

Does the SegNet implementation use Max Pooling Indices? #15

Open Erotemic opened 6 years ago

Erotemic commented 6 years ago

Does the implementation of SegNet use cached max pooling indices for fast upsampling as described in the paper? Looking at the implementation it seems that there is only a custom implementation of a decoder layer, but I don't see how it uses indices to upsample. It seems like it just learns the transpose convolution (deconvolution) layer just as a fully convolutional network would.

Is there something I'm missing?

zijundeng commented 6 years ago

My implementation of SegNet does not use max pooling indices.