中文

LightStereo:仅需通道提升即可实现高效2D代价聚合

计算机视觉与模式识别 2025-02-27 v3

摘要

我们提出了LightStereo,这是一个专为加速配对过程而设计的前沿立体匹配网络。与传统方法依赖计算密集型4D代价聚合不同,LightStereo采用3D代价体作为轻量级替代方案。虽然类似的方法已有探索,但我们的突破性贡献在于通过专注于3D代价体的通道维度来提升性能——该维度封装了匹配代价的分布。我们进行了详尽的探索,获得了多种策略来放大关键维度的容量,确保了精度与效率的平衡。我们在 various benchmarks 上与现有SOTA方法进行了比较,结果显示LightStereo在速度、精度和资源利用方面均表现出色。在SceneFlow数据集上,LightStereo仅需22 GFLOPs和17 ms的运行时间即可实现具竞争力的EPE指标,并在KITTI 2015中位列实时模型榜首。我们的全面分析揭示了2D代价聚合对立体匹配的影响,为高效立体系统的实际应用指明了方向。代码将公开于https://github.com/XiandaGuo/OpenStereo。

关键词

引用

@article{arxiv.2406.19833,
  title  = {LightStereo: Channel Boost Is All You Need for Efficient 2D Cost Aggregation},
  author = {Xianda Guo and Chenming Zhang and Youmin Zhang and Wenzhao Zheng and Dujun Nie and Matteo Poggi and Long Chen},
  journal= {arXiv preprint arXiv:2406.19833},
  year   = {2025}
}

备注

Code will be available at \url{https://github.com/XiandaGuo/OpenStereo}