4 个月前

用于语义分割的全卷积网络

用于语义分割的全卷积网络

摘要

卷积网络是强大的视觉模型,能够生成特征层次结构。我们证明了端到端训练的卷积网络,从像素到像素,能够在语义分割任务中超越之前的最佳结果。我们的关键洞察是构建“全卷积”网络,该网络可以接受任意大小的输入,并高效地进行推理和学习,产生相应大小的输出。我们定义并详细描述了全卷积网络的空间,解释了它们在空间密集预测任务中的应用,并探讨了与先前模型的联系。我们将当代的分类网络(如 AlexNet、VGG 网络和 GoogLeNet)转换为全卷积网络,并通过微调将这些网络学到的表示迁移到分割任务上。随后,我们定义了一种跳跃架构,该架构结合了深层、粗糙层的语义信息和浅层、精细层的外观信息,以生成准确且详细的分割结果。我们的全卷积网络在 PASCAL VOC(2012 年数据集上的平均交并比相对提高了 30%,达到 67.2%)、NYUDv2、SIFT Flow 和 PASCAL-Context 数据集上实现了改进的分割效果,而对一幅典型图像进行推理仅需十分之一秒。

代码仓库

rickyHong/FCN-segmentation-repl
caffe2
GitHub 中提及
shelhamer/fcn.berkeleyvision.org
caffe2
GitHub 中提及
ChenMicky/FCN
tf
GitHub 中提及
geodekid/FCN
caffe2
GitHub 中提及
kevinddchen/Keras-FCN
tf
GitHub 中提及
Lxrd-AJ/Advanced_ML
pytorch
GitHub 中提及
Osdel/ssnets
tf
GitHub 中提及
fmahoudeau/fcn
tf
GitHub 中提及
cooparation/FCN_play
GitHub 中提及
AzogDefiler/Sandbox
GitHub 中提及
shekkizh/FCN.tensorflow
tf
GitHub 中提及
TejasBajania/Mtech_pro
GitHub 中提及

基准测试

基准方法指标
scene-segmentation-on-sun-rgbdFCN
Mean IoU: 27.39
semantic-segmentation-on-cityscapesFCN
Mean IoU (class): 65.3%
semantic-segmentation-on-nyu-depth-v2FCN-32s RGB-HHA
Mean Accuracy: 44
semantic-segmentation-on-pascal-voc-2011-testFCN-pool4
Mean IoU: 22.4
semantic-segmentation-on-pascal-voc-2011-testFCN-VGG16
Mean IoU: 32
video-semantic-segmentation-on-cityscapes-valFCN-50 [14]
mIoU: 70.1

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
用于语义分割的全卷积网络 | 论文 | HyperAI超神经