论文原文
SegNet: A Deep ConvolutionalEncoder-Decoder Architecture for ImageSegmentation
0. 简介
SegNet与DeconvNet在结构上非常相似,也都使用了maxpooling的索引来进行上采样,两者主要区别在于:
- SegNet去掉了FC层,而DeconvNet没有;
- SegNet在decoder中使用了卷积,而DeconvNet使用了反卷积;
去掉了FC层,使得SegNet的参数量更小,也更容易实现端到端的训练,同时训练速度和推理速度都很快。
同时SegNet实现了几乎完全对称encoder-decoder结构。
1. 网络架构
和DeconvNet相同,SegNet的encoder部分基于VGG 16,但是没有卷积层,在进行maxpooling时,同时保存pooling的索引,以便decoder进行上采样