4 个月前

基于CNN的信心传播用于引导稀疏深度回归

基于CNN的信心传播用于引导稀疏深度回归

摘要

通常,卷积神经网络(CNNs)处理的是规则网格上的数据,例如普通相机生成的数据。设计适用于稀疏且不规则分布输入数据的CNN仍然是一个开放的研究问题,其在自动驾驶、机器人技术和监控等领域具有广泛的应用前景。本文提出了一种代数约束归一化卷积层,该层适用于高稀疏度的输入数据,并且相比相关工作具有较少的网络参数。我们提出了新的策略来从卷积操作中确定置信度,并将其传递到后续层。同时,我们还提出了一种目标函数,该函数能够在最小化数据误差的同时最大化输出置信度。为了整合结构信息,我们研究了融合策略,以在归一化卷积网络框架中结合深度和RGB信息。此外,我们引入了输出置信度作为辅助信息来提高结果的质量。我们的归一化卷积网络框架在场景深度补全问题上的能力得到了验证。我们在KITTI-Depth和NYU-Depth-v2数据集上进行了全面的实验。实验结果清楚地表明,所提出的方法在仅需约1-5%的参数数量的情况下,实现了优于现有最先进方法的性能。

代码仓库

abdo-eldesokey/nconv
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
depth-completion-on-kitti-depth-completionNConv-CNN-L2
MAE: 233
RMSE: 830
Runtime [ms]: 20
depth-completion-on-kitti-depth-completionNConv-CNN
MAE: 360
RMSE: 1268
Runtime [ms]: 10
depth-completion-on-kitti-depth-completionNConv-CNN-L1
MAE: 208
RMSE: 859
Runtime [ms]: 20

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于CNN的信心传播用于引导稀疏深度回归 | 论文 | HyperAI超神经