3 个月前

三种利用自监督深度估计提升语义分割的方法

三种利用自监督深度估计提升语义分割的方法

摘要

训练深度神经网络进行语义分割需要大量标注的训练数据,这在实际应用中构成了重大挑战,因为标注分割掩码是一项高度耗时的人工工作。为解决这一问题,我们提出了一种半监督语义分割框架,该框架通过利用未标注图像序列的自监督单目深度估计进行增强。具体而言,本文提出三项关键贡献:(1)将自监督深度估计过程中学习到的特征知识迁移至语义分割任务;(2)基于场景几何信息,通过图像与标签的融合实现强数据增强;(3)在学生-教师框架中,利用深度特征的多样性以及深度学习的难易程度,选择最具价值的样本进行人工标注,以提升语义分割的训练效率。我们在Cityscapes数据集上对所提模型进行了验证,结果表明三个模块均带来了显著的性能提升,并在半监督语义分割任务中达到了当前最优水平。代码实现已开源,地址为:https://github.com/lhoyer/improving_segmentation_with_selfsupervised_depth。

代码仓库

基准测试

基准方法指标
semi-supervised-semantic-segmentation-on-1SegSDE (MTL decoder with ResNet101, ImageNet pretrained, unlabeled image sequences)
Validation mIoU: 69.38%
semi-supervised-semantic-segmentation-on-2SegSDE (MTL decoder with ResNet101, ImageNet pretrained, unlabeled image sequences)
Validation mIoU: 68.01%
semi-supervised-semantic-segmentation-on-3SegSDE (MTL decoder with ResNet101, ImageNet pretrained, unlabeled image sequences)
Validation mIoU: 62.09%

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
三种利用自监督深度估计提升语义分割的方法 | 论文 | HyperAI超神经