深度学习论文: LEDnet: A lightweight encoder-decoder network for real-time semantic segmentation及其PyTorch实现
LEDnet: A lightweight encoder-decoder network for real-time semantic segmentation
PDF:https://arxiv.org/pdf/1905.02423.pdf
PyTorch: https://github.com/shanglianlm0525/PyTorch-Networks
1 概述
LEDNet的不对称结构(asymmetrical architecture),如上图所示,使得网络参数大大减少,加速了推理过程;
残差网络中的 Channel split and shuffle 有强大的特征表示。
在 decoder 端,采用特征金字塔的注意力机制来设计APN,进一步降低了整个网络的复杂性。
模型参数不到1M,并且能够在单个GTX 1080Ti GPU中以超过71 FPS的速度运行。