← 返回日报
🌐 机器翻译 · DeepSeek · GitHub

Robbyant lingbot-map


好的,这是您要求的英文文章的中文翻译,已按照规范处理为 Markdown 格式。

Robbyant lingbot-map

LingBot-Map: Geometric Context Transformer for Streaming 3D Reconstruction

Robbyant Team teaser.mp4

🗺️ 认识 LingBot-Map!我们构建了一个用于流式 3D 重建的前馈式 3D 基础模型!🏗️🌍

LingBot-Map 专注于:

📑 目录

点击展开

📰 新闻

📋 待办事项

⚙️ 安装

📦 模型下载

🚀 快速开始

🎬 交互式演示 (demo.py)

🎥 离线渲染管线 (demo_render/batch_demo.py)

📜 许可证

📖 引用

致谢

📰 新闻

📋 待办事项

⚙️ 安装

  1. 创建 conda 环境
  2.     conda create -n lingbot-map python=3.10 -y
        conda activate lingbot-map
  3. 安装 PyTorch (CUDA 12.8)
  4.     pip install torch==2.8.0 torchvision==0.23.0 --index-url https://download.pytorch.org/whl/cu128

PyTorch 2.8.0 是推荐版本,因为 NVIDIA Kaolin(批处理渲染管线所需)有针对 torch-2.8.0_cu128 的预编译 wheel。如果您只需要 demo.py,可以使用更新的 PyTorch,但批处理渲染器则需要从源码编译 Kaolin。对于其他 CUDA 版本,请参阅 PyTorch 入门指南。

  1. 安装 lingbot-map
  2.     pip install -e .
  3. 安装 FlashInfer(推荐)

FlashInfer 提供分页 KV 缓存注意力,用于高效的流式推理。它是一个纯 Python 包,在首次使用时通过 JIT 编译 CUDA 内核,因此单个 wheel 可以跨 CUDA/PyTorch 版本使用:

    pip install --index-url https://pypi.org/simple flashinfer-python

--index-url https://pypi.org/simple 仅在您的默认 pip 索引是没有 flashinfer-python 的内部镜像时才需要。

(可选)为了加快首次使用速度,您还可以安装特定 CUDA 版本的 JIT 缓存:

    pip install flashinfer-jit-cache -f https://flashinfer.ai/whl/cu128/flashinfer-jit-cache/

详情请参阅 FlashInfer 安装指南。

如果未安装 FlashInfer,模型将通过 --use_sdpa 回退到 SDPA(PyTorch 原生注意力)。

  1. 可视化依赖项(可选)
  2.     pip install -e ".[vis]"

📦 模型下载

| 模型名称 | Huggingface 仓库 | ModelScope 仓库 | 描述 | | :--- | :--- | :--- | :--- | | lingbot-map-long | robbyant/lingbot-map | Robbyant/lingbot-map | 更适合长序列和大规模场景。 | | lingbot-map | robbyant/lingbot-map | Robbyant/lingbot-map | 平衡检查点(论文、基准测试和离线演示中使用)——在短序列和长序列之间权衡综合性能。 | | lingbot-map-stage1 | robbyant/lingbot-map | Robbyant/lingbot-map | lingbot-map 的阶段一训练检查点——可以加载到 VGGT 模型中进行双向推理 (c2w)。 |

🚧 即将推出:我们正在训练一个支持更长序列的更强模型——敬请期待。

🚀 快速开始

安装完成后,使用一条命令运行您的第一个场景:

python demo.py --model_path /path/to/lingbot-map-long.pt \
               --image_folder example/courthouse --mask_sky

这将启动一个交互式 viser 查看器,地址为 http://localhost:8080。请参阅下面的“交互式演示”以获取完整的场景和参数列表,或跳转到“离线渲染管线”以进行长序列批处理渲染。

🎬 交互式演示 (demo.py)

运行 demo.py 可通过基于浏览器的 viser 查看器(默认 http://localhost:8080)进行交互式 3D 可视化。

尝试示例场景

我们在 example/ 中提供了四个示例场景,您可以开箱即用:

📖 阅读原文 →