深度估计 DenseDepth 笔记

本文详细介绍了DenseDepth方法,该方法利用预训练的DenseNet-169作为编码器,结合shortcut链接和上采样模块构建U型网络。通过深度、梯度和SSIM损失函数进行监督学习,对深度图进行回归。文章对比了与其他方法(如DRON)的差异,并讨论了深度归一化的应用。
摘要由CSDN通过智能技术生成

参考代码:DenseDepth
论文名称:High Quality Monocular Depth Estimation via Transfer Learning

1. 概述

导读:这篇文章是使用深度监督的方式进行训练的,文章使用在ImageNet上训练过的DenseNet-169作为编码器,之后使用shortcut链接和上采样模块(上采样操作+2个卷积层)作为解码器,从而构建一个U型的网络结构。在此基础上使用输入的深度GT作为回归目标,从而实现网络的训练。

在下图中可以将文章的方法与DRON方法(分段回归)进行比较,文章的方法更加自然和实际。
在这里插入图片描述

2. 方法设计

2.1 网络结构

文章的网络结构比较简单,是一个带shortcut链接的U型网络结构:
在这里插入图片描述
编码器是DenseNet-169(pretrained on ImageNet),解码器由上采样模块(2x上采样+2个不带BN的卷积)构成,之后在解码器的最后一个上采样模块上预测对应的深度图。

2.2 损失函数

预测得到的深度图表示为 y ^ \hat{y} y^,对应的标注GT表示为 y y y,那么预测与回归之间的监督关系被描述为:
L ( y , y ^ ) = λ L d e p t h ( y , y ^ ) + L g r a d ( y , y ^ ) + L S S I M ( ( y , y ^ ) ) L(y,\hat{y})=\lambda L_{depth}(y,\hat{y})+L_{grad}(y,\hat{y})+L_{SSIM}((y,\hat{y})) L(y,y^)=λLdepth(y,y^)+Lgrad(y

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值