4 个月前

基于区域的语义分割与端到端训练

基于区域的语义分割与端到端训练

摘要

我们提出了一种新的语义分割方法,该方法的任务是对图像中的每个像素进行语义类别标注。我们的方法结合了两种主要竞争范式的优点。基于区域分类的方法为外观测量提供了适当的空域支持,但通常分为两个独立的阶段,其中任何一个阶段都不是为了最终的像素标注性能而设计的。最近的全卷积方法能够实现端到端训练以完成最终的像素标注任务,但依赖于固定的补丁作为空域支持。我们展示了如何修改现代基于区域的方法,以实现语义分割的端到端训练。这是通过一个可微分的区域到像素层和一个可微分的自由形状感兴趣区域(Region-of-Interest, RoI)池化层来实现的。我们的方法在类平均精度方面取得了显著改进,在SIFT Flow数据集上达到了64.0%,在PASCAL Context数据集上达到了49.9%,并且在物体边界处特别准确。

代码仓库

nightrome/matconvnet-calvin
官方
GitHub 中提及

基准测试

基准方法指标
semantic-segmentation-on-pascal-contextRBE2E
Mean Accuracy: 49.9
Pixel Accuracy: 62.4
mIoU: 32.5
semantic-segmentation-on-sift-flowRBE2E
Mean Accuracy: 64
Pixel Accuracy: 84.3

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于区域的语义分割与端到端训练 | 论文 | HyperAI超神经