VQT-Light:具有更丰富纹理的轻量级 HDR 光照图预测
计算机视觉与模式识别
2025-09-17 v1
摘要
准确的光照估计是计算机视觉和图形学中一项重要且具有挑战性的任务。然而,现有方法要么难以恢复光照图的详细纹理,要么在运行速度和纹理保真度方面面临挑战。为了解决这个问题,我们提出了一种基于 VQVAE 和 ViT 架构的新框架。VQT-Light 包含两个模块:特征提取和光照估计。首先,我们利用 VQVAE 提取光照图的离散特征而非连续特征,以避免“后验崩塌”。其次,我们通过 ViT 而非 CNN 来捕获输入图像的全局上下文和依赖关系,以改善视场外光照的预测。结合上述两个模块,我们将光照估计表述为多类分类任务,这在我们的流程中起着关键作用。因此,我们的模型在保持轻量级和快速的同时,预测出具有更丰富纹理和更好保真度的光照图。VQT-Light 实现了 40FPS 的推理速度,并改善了多个评估指标。定性和定量实验表明,与现有 SOTA 方法相比,所提方法取得了更优的结果。
引用
@article{arxiv.2509.12556,
title = {VQT-Light:Lightweight HDR Illumination Map Prediction with Richer Texture.pdf},
author = {Kunliang Xie},
journal= {arXiv preprint arXiv:2509.12556},
year = {2025}
}
备注
11 pages, 8 figures