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

DiffPoint:基于 ViT 扩散模型的单视图与多视图点云重建

DiffPoint: Single and Multi-view Point Cloud Reconstruction with ViT Based Diffusion Model

Fudan University(复旦大学);Alibaba Group(阿里巴巴集团)

Yu Feng, Xing Shi, Mengli Cheng, Yun Xiong

AI总结:

提出 DiffPoint 架构,结合 ViT 与扩散模型,通过将含噪点云划分为不规则块并作为 token 处理,在单视图和多视图点云重建任务上取得最先进结果,并引入灵活特征融合模块,验证了统一架构改进 3D 重建的可行性。

摘要:

随着 2D 到 3D 重建任务在各种现实场景中获得重要关注,生成高质量点云变得至关重要。尽管深度学习模型近期在生成点云方面取得了成功,但由于图像与点云之间的差异,生成高保真结果仍面临挑战。虽然视觉 Transformer(ViT)和扩散模型在各种视觉任务中展现出潜力,但它们在从图像重建点云方面的优势尚未得到证明。在本文中,我们首先提出了一种简洁且强大的架构 DiffPoint,将 ViT 和扩散模型结合用于点云重建任务。在每个扩散步骤中,我们将含噪点云划分为不规则块。然后,使用将所有输入视为 token(包括时间信息、图像嵌入和含噪块)的标准 ViT 主干网络,我们训练模型基于输入图像预测目标点。我们在单视图和多视图重建任务上评估了 DiffPoint,并取得了最先进的结果。此外,我们引入了一个统一且灵活的特征融合模块,用于聚合来自单个或多个输入图像的图像特征。进一步地,我们的工作证明了跨语言和图像应用统一架构以改进 3D 重建任务的可行性。

英文摘要:

As the task of 2D-to-3D reconstruction has gained significant attention in various real-world scenarios, it becomes crucial to be able to generate high-quality point clouds. Despite the recent success of deep learning models in generating point clouds, there are still challenges in producing high-fidelity results due to the disparities between images and point clouds. While vision transformers (ViT) and diffusion models have shown promise in various vision tasks, their benefits for reconstructing point clouds from images have not been demonstrated yet. In this paper, we first propose a neat and powerful architecture called DiffPoint that combines ViT and diffusion models for the task of point cloud reconstruction. At each diffusion step, we divide the noisy point clouds into irregular patches. Then, using a standard ViT backbone that treats all inputs as tokens (including time information, image embeddings, and noisy patches), we train our model to predict target points based on input images. We evaluate DiffPoint on both single-view and multi-view reconstruction tasks and achieve state-of-the-art results. Additionally, we introduce a unified and flexible feature fusion module for aggregating image features from single or multiple input images. Furthermore, our work demonstrates the feasibility of applying unified architectures across languages and images to improve 3D reconstruction tasks.

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

↑