中文

ThermoStereoRT:基于知识蒸馏与注意力细化的实时热红外立体匹配

计算机视觉与模式识别 2025-04-11 v1

摘要

本文提出ThermoStereoRT,一种面向全天候条件的实时热红外立体匹配方法,可从两幅已校正的热红外立体图像中恢复视差,适用于夜间无人机监控或床下清洁机器人等应用场景。ThermoStereoRT采用轻量而强大的骨干网络,从热红外图像构建三维代价体,并利用多尺度注意力机制生成初始视差图。为细化该视差图,设计了一种新颖的通道与空间注意力模块。针对热红外图像中地面真值数据稀疏的挑战,利用知识蒸馏来提升性能且不增加计算开销。在多个数据集上的全面评估表明,ThermoStereoRT兼具实时处理能力和鲁棒的准确性,使其成为各种挑战性环境中实际部署的有前景的解决方案。代码将发布于 https://github.com/SJTU-ViSYS-team/ThermoStereoRT

关键词

引用

@article{arxiv.2504.07418,
  title  = {ThermoStereoRT: Thermal Stereo Matching in Real Time via Knowledge Distillation and Attention-based Refinement},
  author = {Anning Hu and Ang Li and Xirui Jin and Danping Zou},
  journal= {arXiv preprint arXiv:2504.07418},
  year   = {2025}
}

备注

7 pages, 6 figures, 4 tables. Accepted to IEEE ICRA 2025. This is the preprint version