Computer Science > Computer Vision and Pattern Recognition
A newer version of this paper has been withdrawn by Juntang Zhuang
[Submitted on 27 Nov 2018 (v1), revised 10 Dec 2018 (this version, v2), latest version 4 Sep 2019 (v6)]
Title:ShelfNet for Real-time Semantic Segmentation
No PDF available, click to view other formatsAbstract:In this project, we present ShelfNet, a lightweight convolutional neural network for accurate real-time semantic segmentation. Different from the standard encoder-decoder structure, ShelfNet has multiple encoder-decoder branch pairs with skip connections at each spatial level, which looks like a shelf with multiple columns. The shelf-shaped structure provides multiple paths for information flow and improves segmentation accuracy. Inspired by the success of recurrent convolutional neural networks, we use modified residual blocks where two convolutional layers share weights. The shared-weight block enables efficient feature extraction and model size reduction. We tested ShelfNet with ResNet50 and ResNet101 as the backbone respectively: they achieved 59 FPS and 42 FPS respectively on a GTX 1080Ti GPU with a 512x512 input image. ShelfNet achieved high accuracy: on PASCAL VOC 2012 test set, it achieved 84.2% mIoU with ResNet101 backbone and 82.8% mIoU with ResNet50 backbone; it achieved 75.8% mIoU with ResNet50 backbone on Cityscapes dataset. ShelfNet achieved both higher mIoU and faster inference speed compared with state-of-the-art real-time semantic segmentation models. We provide the implementation this https URL.
Submission history
From: Juntang Zhuang [view email][v1] Tue, 27 Nov 2018 20:57:48 UTC (3,415 KB)
[v2] Mon, 10 Dec 2018 20:39:06 UTC (1 KB) (withdrawn)
[v3] Wed, 12 Dec 2018 02:24:06 UTC (1 KB) (withdrawn)
[v4] Thu, 13 Dec 2018 09:51:49 UTC (3,411 KB)
[v5] Sat, 15 Dec 2018 14:13:15 UTC (1 KB) (withdrawn)
[v6] Wed, 4 Sep 2019 03:59:15 UTC (3,883 KB)
References & Citations
Bibliographic and Citation Tools
Bibliographic Explorer (What is the Explorer?)
Litmaps (What is Litmaps?)
scite Smart Citations (What are Smart Citations?)
Code, Data and Media Associated with this Article
CatalyzeX Code Finder for Papers (What is CatalyzeX?)
DagsHub (What is DagsHub?)
Gotit.pub (What is GotitPub?)
Papers with Code (What is Papers with Code?)
ScienceCast (What is ScienceCast?)
Demos
Recommenders and Search Tools
Influence Flower (What are Influence Flowers?)
Connected Papers (What is Connected Papers?)
CORE Recommender (What is CORE?)
arXivLabs: experimental projects with community collaborators
arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.
Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.
Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.