4 个月前

DFANet:深度特征聚合用于实时语义分割

DFANet:深度特征聚合用于实时语义分割

摘要

本文介绍了一种在资源受限条件下用于语义分割的极其高效的卷积神经网络架构,名为DFANet。我们提出的网络从一个轻量级主干网络开始,通过子网络和子阶段级联分别聚合判别特征。基于多尺度特征传播,DFANet显著减少了参数数量,但仍获得了足够的感受野并增强了模型的学习能力,从而在速度和分割性能之间达到了平衡。在Cityscapes和CamVid数据集上的实验表明,DFANet的浮点运算量(FLOPs)比现有的最先进的实时语义分割方法减少了8倍,同时速度提高了2倍,而精度相当。具体而言,在单个NVIDIA Titan X显卡上,DFANet仅需1.7 GFLOPs即可实现每秒160帧的速度,并在Cityscapes测试数据集上达到70.3%的平均交并比(Mean IOU)。当处理更高分辨率的图像时,其浮点运算量为3.4 GFLOPs,平均交并比提高到71.3%。

代码仓库

j-a-lin/DFANet_PyTorch
pytorch
GitHub 中提及

基准测试

基准方法指标
semantic-segmentation-on-camvidDFANet A
Mean IoU: 64.7%
semantic-segmentation-on-cityscapesDFANet A
Mean IoU (class): 71.3%
smac-on-smac-def-infantry-parallelDIQL
Median Win Rate: 45.0

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
DFANet:深度特征聚合用于实时语义分割 | 论文 | HyperAI超神经