2 个月前

ShelfNet用于快速语义分割

Juntang Zhuang; Junlin Yang; Lin Gu; Nicha Dvornek
ShelfNet用于快速语义分割
摘要

本文介绍了一种名为ShelfNet的新颖架构,用于实现快速且准确的语义分割。与单一的编码器-解码器结构不同,ShelfNet在每个空间层次上都具有多个带有跳过连接的编码器-解码器分支对,其结构类似于多列货架。这种货架状结构可以视为多个深浅路径的集成,从而提高了准确性。我们通过减少通道数量显著降低了计算负担,同时凭借这一独特结构实现了高精度。此外,我们在残差块中提出了一种共享权重策略,该策略在不牺牲性能的前提下减少了参数数量。与PSPNet等流行的非实时方法相比,我们的ShelfNet在PASCAL VOC数据集上以相似的精度实现了4倍更快的推理速度。与BiSeNet等实时分割模型相比,我们的模型在Cityscapes数据集上以相当的速度实现了更高的精度,适用于自动驾驶中的街景理解等对速度有较高要求的任务。此外,我们的ShelfNet使用ResNet34作为骨干网络,在Cityscapes数据集上达到了79.0%的mIoU(平均交并比),优于使用较大骨干网络如ResNet101的PSPNet和BiSeNet。通过广泛的实验验证,我们证明了ShelfNet的优越性能。我们提供了实现代码的链接:\url{https://github.com/juntang-zhuang/ShelfNet-lw-cityscapes}。

ShelfNet用于快速语义分割 | 最新论文 | HyperAI超神经