
摘要
编码器-解码器框架在离线语义图像分割领域处于领先地位。随着自主系统的兴起,实时计算变得越来越重要。本文介绍了一种快速分割卷积神经网络(Fast-SCNN),该模型能够在高分辨率图像数据(1024x2048像素)上实现超过实时的语义分割,并适用于低内存嵌入式设备上的高效计算。基于现有的双分支快速分割方法,我们引入了“学习降采样”模块,该模块能够同时为多个分辨率分支计算低级特征。我们的网络结合了高分辨率下的空间细节和低分辨率下提取的深层特征,从而在Cityscapes数据集上实现了每秒123.5帧、平均交并比为68.0%的性能。我们还证明了大规模预训练并非必要。通过使用ImageNet预训练和Cityscapes粗标签数据进行实验,我们对这一结论进行了充分验证。最后,我们在不修改网络结构的情况下,展示了在子采样输入上实现更快计算且具有竞争力的结果。
代码仓库
SkyWa7ch3r/ImageSegmentation
tf
GitHub 中提及
PaddlePaddle/PaddleSeg
paddle
dbaofd/solar-panels-detection
tf
GitHub 中提及
JonasFrey96/ASL
pytorch
GitHub 中提及
Tramac/Fast-SCNN-pytorch
pytorch
GitHub 中提及
rachthree/fast_scnn
tf
GitHub 中提及
zh320/realtime-semantic-segmentation-pytorch
pytorch
GitHub 中提及
SkyWa7ch3r/SceneSegmentation
tf
GitHub 中提及
kshitizrimal/Fast-SCNN
tf
GitHub 中提及
osmr/imgclsmob
mxnet
GitHub 中提及
open-mmlab/mmsegmentation
pytorch
VilledeMontreal/urban-segmentation
pytorch
GitHub 中提及
rudolfsteiner/fast_scnn
tf
GitHub 中提及
Eladamar/fast_scnn
pytorch
GitHub 中提及
Mind23-2/MindCode-44
mindspore
GitHub 中提及
ayushmankumar7/Fast-Segementation
tf
GitHub 中提及
minus31/Portrait-segmentation
tf
GitHub 中提及
DeepVoltaire/Fast-SCNN
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| semantic-segmentation-on-cityscapes | Fast-SCNN | Mean IoU (class): 68% |
| semantic-segmentation-on-cityscapes-val | Fast-SCNN + Coarse + ImageNet | mIoU: 69.19 |
| semantic-segmentation-on-dada-seg | Fast-SCNN | mIoU: 26.32 |
| semantic-segmentation-on-densepass | Fast-SCNN | mIoU: 24.6% |
| semantic-segmentation-on-eventscape | Fast-SCNN | mIoU: 44.27 |
| semantic-segmentation-on-synpass | Fast-SCNN | mIoU: 21.30% |
| thermal-image-segmentation-on-pst900 | Fast-SCNN | mIoU: 47.2 |