arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2404.05505cs.CVcs.LGcs.RO

驯服 Transformer 以实现逼真的激光雷达点云生成

Taming Transformers for Realistic Lidar Point Cloud Generation

WMG, University of Warwick(华威大学 WMG)

Hamed Haghighi, Amir Samadi, Mehrdad Dianati, Valentina Donzella, Kurt Debattista

AI总结:

本文提出 LidarGRIT,通过自回归 Transformer 在 VQ-VAE 潜在空间迭代采样距离图像并分别解码射线丢失掩码,解决扩散模型难以逼真建模激光雷达射线丢失噪声的问题,在两个 KITTI 数据集上超越 SOTA。

摘要:

扩散模型(Diffusion Models,DMs)凭借稳定的训练以及采样过程中的迭代细化,已在激光雷达点云生成任务中取得最先进(State-Of-The-Art,SOTA)结果。然而,由于其固有的去噪过程,DMs 往往无法逼真地建模激光雷达射线丢失噪声。为在保留迭代采样优势的同时增强射线丢失噪声的生成,我们提出 LidarGRIT,这是一种使用自回归 transformer 在潜在空间而非图像空间中迭代采样距离图像的生成模型。此外,LidarGRIT 利用 VQ-VAE 分别解码距离图像和射线丢失掩码。我们的结果表明,在 KITTI-360 和 KITTI odometry 数据集上,LidarGRIT 相比 SOTA 模型取得了更优性能。代码可在 https://github.com/hamedhaghighi/LidarGRIT 获取。

英文摘要:

Diffusion Models (DMs) have achieved State-Of-The-Art (SOTA) results in the Lidar point cloud generation task, benefiting from their stable training and iterative refinement during sampling. However, DMs often fail to realistically model Lidar raydrop noise due to their inherent denoising process. To retain the strength of iterative sampling while enhancing the generation of raydrop noise, we introduce LidarGRIT, a generative model that uses auto-regressive transformers to iteratively sample the range images in the latent space rather than image space. Furthermore, LidarGRIT utilises VQ-VAE to separately decode range images and raydrop masks. Our results show that LidarGRIT achieves superior performance compared to SOTA models on KITTI-360 and KITTI odometry datasets. Code available at:https://github.com/hamedhaghighi/LidarGRIT.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑