HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

DSNet: A Novel Way to Use Atrous Convolutions in Semantic Segmentation

Zilu Guo; Liuyang Bian; Xuan Huang; Hu Wei; Jingyu Li; Huasheng Ni

DSNet: A Novel Way to Use Atrous Convolutions in Semantic Segmentation

Abstract

Atrous convolutions are employed as a method to increase the receptive field in semantic segmentation tasks. However, in previous works of semantic segmentation, it was rarely employed in the shallow layers of the model. We revisit the design of atrous convolutions in modern convolutional neural networks (CNNs), and demonstrate that the concept of using large kernels to apply atrous convolutions could be a more powerful paradigm. We propose three guidelines to apply atrous convolutions more efficiently. Following these guidelines, we propose DSNet, a Dual-Branch CNN architecture, which incorporates atrous convolutions in the shallow layers of the model architecture, as well as pretraining the nearly entire encoder on ImageNet to achieve better performance. To demonstrate the effectiveness of our approach, our models achieve a new state-of-the-art trade-off between accuracy and speed on ADE20K, Cityscapes and BDD datasets. Specifically, DSNet achieves 40.0% mIOU with inference speed of 179.2 FPS on ADE20K, and 80.4% mIOU with speed of 81.9 FPS on Cityscapes. Source code and models are available at Github: https://github.com/takaniwa/DSNet.

Code Repositories

takaniwa/dsnet
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
real-time-semantic-segmentation-on-cityscapes-1DSNet
Frame (fps): 81.9
mIoU: 80.4%
semantic-segmentation-on-bdd100k-valDSNet-Base
mIoU: 64.6
semantic-segmentation-on-bdd100k-valDSNet-head64
mIoU: 62.6(172.2FPS 4090)
semantic-segmentation-on-camvidDSNet-Base
Mean IoU: 83.32
semantic-segmentation-on-cityscapes-valDSNet(single-scale)
FPS: 81.9
mIoU: 80.4
semantic-segmentation-on-cityscapes-valDSNet-Base(single-scale)
mIoU: 82.0

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
DSNet: A Novel Way to Use Atrous Convolutions in Semantic Segmentation | Papers | HyperAI