Skip to content

tianxingleo/BrainDance

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1,269 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

BrainDance Logo

BrainDance | 流光 · 记

“物理世界注定走向无序,而我们在比特世界重建永恒。”

面向空间记忆的三维语义引擎与检索系统。

English | 简体中文

从采集到检索的三维记忆链路

An Anti-Entropy Engine for Human Memory

Flutter Supabase Python Vue PostgreSQL pgvector Three.js WebXR Deno License: PolyForm Noncommercial

项目概述 (Overview)

BrainDance (流光 · 记) 是一个把真实空间保存为可浏览、可检索、可回访数字资产的项目。

它的出发点很简单:照片和视频能留下画面,但很难保留空间感。一个房间怎么摆、一个街角从哪看过去、某件东西当时放在什么位置,这些信息在 2D 媒介里通常会被压缩掉。

BrainDance 用 3D Gaussian Splatting (高斯泼溅) 等技术把真实场景重建成三维资产,再接入 Multimodal AI (多模态大模型)RAG (检索增强生成),让这些场景不只是“能看”,还可以被搜索、定位和回溯。可以把它理解成一个面向现实空间的记忆索引系统。

核心特性

  • 📷 低门槛采集:用手机视频或图片就能发起空间采集,重建计算放在云端完成。
  • 🔍 空间语义检索:结合多模态理解、向量检索和空间锚点,在三维场景里查找物体、位置和相关视角。
  • ⏳ 时间维度回看:围绕同一空间的多次扫描结果,比较不同时间下的变化。
  • 🥽 AR/VR 沉浸式查看:移动端支持基于 MindAR 图像识别的 Marker AR 模式,将 3DGS 模型通过打印纸板锚定到真实环境,支持缩放、旋转和高度调节;桌面端提供独立 VR 渲染器(vr-3dgs-viewer),基于 WebXR 标准支持 SteamVR 头显中的抓取缩放、导航跳转与空间测量。
  • ☁️ 端云协同链路:移动端负责采集与查看,Supabase 和 AI Worker 负责存储、调度与重建处理。

技术栈速览

前端与交互

Flutter Riverpod WebView Vue 3 Vite TypeScript Three.js WebXR ECharts

云服务与数据

Supabase PostgreSQL pgvector Deno Docker

AI 与三维重建

Python PyTorch FFmpeg COLMAP GLOMAP Nerfstudio Qwen-VL

为什么做这个项目

BrainDance 面向的是一个很具体的移动端问题:手机已经能高频记录生活,但现有记录方式大多停留在照片、视频和文字层面,真正和空间有关的信息很难被保存和再次利用。

照片和视频当然有价值,但它们记录的大多只是某个角度下的一帧。过几年再回看,你可能还记得画面,却很难重新找回那个空间本身的尺度、方向、布局,以及某件物品具体放在哪里。

我们希望把这件事做成一个真正能在手机场景里使用的应用链路:用户在移动端完成采集,云端完成三维重建与语义理解,再把结果返回到移动端和 Web 端进行浏览、检索和回看。项目使用 3D Gaussian Splatting (高斯泼溅)Multimodal AI (多模态大模型)RAG (检索增强生成),目标不是单纯生成一个 3D 模型,而是把现实空间整理成可检索、可定位、可追溯的数字记忆资产。

应用价值:个人记忆、公共档案与长期留存

结合比赛强调的移动端创新、真实场景和落地能力,BrainDance 当前主要面向三类场景:

1. 微观尺度 (The Micro Scale)

  • 个人见证 (Spatial Journal)
    • 比如毕业离开住了 4 年的宿舍,或者搬离住过很久的出租屋,用户可以在离开前用手机完成一次采集,把整个空间保存下来,之后再回看时看到的不只是一张照片,而是一个还能重新进入和检索的场景。
  • 医疗辅助 (The Cure)
    • 在阿尔茨海默症等记忆障碍场景里,空间回访可能比平面照片更接近人对熟悉环境的真实感受,这类能力也更符合移动端随时采集、随时调取的使用方式。

2. 宏观尺度 (The Macro Scale)

  • 众包档案馆 (Crowd-Sourced Archive)
    • 如果能持续汇聚用户扫描数据,就有机会把正在消失的街区、店铺和建筑保留下来,形成可进入、可浏览、可检索的三维档案。
  • 集体记忆 (Collective Memory)
    • 这样留下来的内容,不只是文字和图片,而是后续可以真正走进去看、并支持语义检索的空间材料。

3. 时间尺度 (The Temporal Scale)

  • 数字底片 (The Digital Negative)
    • 2D 视频的展示方式很容易过时,但空间资产可以随着终端能力提升继续被重新渲染,在后续设备上仍有复用价值。
  • 面向未来 (Future-Proof)
    • 今天保存下来的不只是影像文件,也可能是未来 XR 设备仍然可以继续使用的空间数据。这让项目具备了从手机应用延展到空间计算场景的潜力。

核心能力与创新点

空间 RAG:像搜索文字一样搜索现实

对 BrainDance 来说,三维重建不是终点。更关键的是把场景里的物体、位置和时间信息整理成可检索的数据,让用户可以像搜索文字一样搜索现实空间。接入 Multimodal LLM (多模态大语言模型) 后,用户不需要手动翻看每个场景,而是可以直接用自然语言发起查询。

  • User Query: "爷爷留下的那块怀表在哪?"
  • System Action: 语义理解 -> 空间索引匹配 -> 摄像机自动飞越 -> 显示最相关的空间位置与视角

时光剥离:在同一空间里回看不同时间

BrainDance 不只想记录一个静态场景,也想记录同一空间在不同时间下的变化。对于房间布置、成长记录、装修过程和城市更新这类场景,在同一坐标系里对比前后变化,往往比按时间顺序翻照片更直观。

Agent Recall:把检索升级成可解释的多轮空间助理

当前仓库已在 supabase/functions/agent-recall 落地统一 Agent 入口,支持空间检索、资产整理、时间对比、记忆图谱等多模式路由,并通过流式协议输出结构化结果(答案、证据、动作、候选),由 Flutter Recall 页直接消费。详细编排能力与协议说明见 语义搜索与 Agent 编排docs/09-LangChain专题/

技术架构与实现 (Technical Architecture)

本项目采用 Supabase BaaS 架构,实现了从移动端采集到云端重建,再到多端检索与浏览的端云协同流程。

系统当前由四个核心部分组成,并通过 Supabase 做任务、数据和状态解耦:

  1. Client (Flutter)
    负责素材采集、上传、任务创建、状态查看与模型浏览。客户端直接连接 Supabase Storage / DB,并通过 Realtime 获取任务进度,是整条用户链路的交互入口。

  2. Backend as a Service (Supabase)
    提供 PostgreSQL、Storage、Auth、Realtime 与 Edge Functions:

    • PostgreSQL + pgvector:存储任务、资产与语义向量。
    • 业务表:当前主链路依赖 processing_tasksmodel_assetsmemory_posescommunity_postsworker_nodes
    • Storage:统一管理原始素材、缩略图、模型文件和相关输出。
    • Realtime:为移动端与 Dashboard 提供状态同步。
    • RLS:基于数据库策略控制用户资产访问权限,并为 Dashboard 提供只读读表策略。
  3. Edge Functions (Deno)
    当前仓库已不只包含基础搜索函数,还形成了“基础检索 + 统一 Agent 入口 + 实验入口”的分层:

    • supabase/functions/search-models:基础语义搜索能力,负责 Embedding、时间解析与向量检索。
    • supabase/functions/agent-recall:正式统一入口,负责多模式路由、流式事件输出、稳定动作协议与前端会话状态。
    • supabase/functions/spatial-search-agent:LangChain / Agent 实验入口,复用共享 Core。
    • supabase/functions/time-compare-agent:面向双时间窗口的专用时间对比能力。
  4. AI Worker (Python)
    部署在 Linux / WSL GPU 节点,监听 processing_tasks,根据 task_type 执行不同流水线,上传结果并回写日志、评分、标签和资产信息,承担项目主要的 AI 和 3D 重建计算。

端侧本地大模型引擎

为了在移动端实现断网环境下的空间检索,项目基于 Qwen3-1.7B 完成了从知识蒸馏、LoRA 微调到 GGUF 量化(3.3GB → 1.2GB)的全链路端侧部署,使主流手机能够在本地运行可靠的语义理解与拒答控制。

完整的模型探索、蒸馏微调流程、量化评估与技术演进记录,请参阅 docs/04-本地问答与微调/ 及其对应的 ai_engine/finetune_qwen3/ 工程模块。

系统组成 (Project Structure)

本项目采用 Monorepo 结构,把移动端、云服务、AI 引擎和可视化工具放在同一仓库中,方便统一开发、联调和演示:

BrainDance/
├── app/                  # [Flutter] 移动端客户端
│   ├── lib/              #   - 登录、录制、Recall、Community、设置等页面与服务
│   ├── assets/           #   - 图标、字体、内置 WebGL 资源
│   └── pubspec.yaml      #   - Flutter 依赖定义
│
├── ai_engine/            # [Python] 核心算法引擎
│   ├── 3dgs/             #   - GPU Worker 与 3D/2D 重建流水线
│   │   ├── src/core/     #       - Worker、工厂、主流程
│   │   ├── src/pipelines/#       - 各类 Pipeline 实现
│   │   ├── src/libs/     #       - nerfstudio / SuGaR / SHARP 等子模块
│   │   ├── tests/        #       - 测试脚本
│   │   ├── requirements.txt
│   │   └── main.py
│   ├── demo/             #   - 演示脚本与实验代码
│   └── finetune_qwen3/   #   - Qwen3 本地问答微调、量化、评测与部署候选验证
│
├── supabase/             # [BaaS] 本地后端基础设施
│   ├── migrations/       #   - SQL 迁移
│   ├── functions/        #   - Edge Functions
│   │   ├── search-models/#       - 基础语义搜索函数
│   │   ├── agent-recall/#        - 统一 Agent Recall 入口
│   │   ├── spatial-search-agent/# - LangChain 试验入口
│   │   ├── time-compare-agent/#  - 双时间窗口对比
│   │   └── _shared/agent-core/#  - 共享 Agent Core 与工具
│   ├── config.toml
│   └── README.md
│
├── dashboard/            # [Vue 3 + Vite] 系统状态看板
│   ├── src/              #   - Dashboard 前端代码
│   ├── .env.example      #   - 环境变量模板
│   └── package.json
│
├── 3dgs_viewer/          # [Tools] 3DGS 查看器与辅助工具
│   ├── my-3dgs-viewer/   #   - Vue 3 桌面端 3DGS 查看器(自由/轨道/电影模式)
│   ├── spark-3dgs-viewer/#   - Spark 查看器前端(含 Marker AR 纸板锚定模式)
│   └── *.py              #   - 位姿导出、图片同步、标签等辅助脚本
├── vr-3dgs-viewer/       # [VR] Vue 3 VR 查看器(Desktop/Stereo/WebXR 三模式)
├── docs/                 # [Doc] 项目文档与技术报告
└── README.md

这套结构对应的是一条完整的软件链路,而不是单一算法仓库:app/ 负责采集与交互,supabase/ 负责存储和任务流转,ai_engine/ 负责重建与理解,dashboard/3dgs_viewer/ 负责观测、调试和展示。

快速开始 (Quick Start)

环境要求 (Prerequisites)

  • AI Engine: NVIDIA GPU,CUDA 12.8,Python 3.10+
  • Hardware: 具体显卡、CPU 与内存要求见下方“测试环境”
  • Infrastructure: Docker, Supabase CLI
  • Client: Flutter SDK, Android Studio / Xcode
  • Dashboard: Node.js 18+

最短启动路径

如果只想快速跑通主链路,按下面的顺序启动即可:

  1. supabase/ 中运行本地服务。
  2. ai_engine/3dgs/ 中启动 Worker。
  3. dashboard/ 中启动状态看板。
  4. app/ 中启动移动端应用并创建任务。

测试环境 (Testing Environment)

移动端测试设备

  • OPPO Find X8

服务器配置

  • 当前 AI Engine 测试/推荐服务器(本机,2026-03-09 实测)
  • CPU: Intel Xeon Platinum 8260 × 2(双路,96 线程)
  • 内存: 503GiB(约 512GB)
  • 显卡: NVIDIA L20 46GB × 2(双卡)
  • 操作系统: Ubuntu 22.04.5 LTS(Kernel 6.8.0-100-generic)

AI Engine 最低配置(兼容基线)

  • CPU: Intel Core i5-14600KF
  • 内存: 64GB RAM
  • 显卡: NVIDIA RTX 5070 12GB

启动步骤 (Runbook)

0. 获取完整代码(含子模块与 LFS)

git lfs install
git clone --recurse-submodules https://github.com/tianxingleo/BrainDance.git
cd BrainDance
git lfs pull
git submodule foreach --recursive 'git lfs pull || true'

1. 启动基础设施 (Supabase Local)

cd supabase
supabase start

启动后记下:

  • API URL
  • anon key
  • service_role key

2. 启动计算引擎 (AI Worker)

conda create -n braindance python=3.10
conda activate braindance
cd ai_engine/3dgs
pip install -r requirements.txt
pip uninstall -y nerfstudio
pip install -e src/libs/nerfstudio
python main.py

如需本地调试单个视频:

cd ai_engine/3dgs
python main.py /path/to/video.mp4

3. 启动 Dashboard

先在 dashboard/.env 中配置:

VITE_SUPABASE_URL=http://127.0.0.1:54321
VITE_SUPABASE_ANON_KEY=YOUR_SUPABASE_ANON_KEY
VITE_STORAGE_BUCKETS=braindance-assets
VITE_SUPABASE_EDGE_FUNCTIONS=search-models,agent-recall,spatial-search-agent,time-compare-agent,test-timeout

然后运行:

cd dashboard
npm install
npm run dev

4. 启动 VR 查看器(可选)

VR 查看器是独立 Vue 3 前端,支持 Desktop / Stereo / WebXR 三种模式:

cd vr-3dgs-viewer
npm install
npm run dev

日常调试可打开 https://127.0.0.1:5174/?preview=desktop 先在普通屏幕上确认模型加载与朝向。启动 SteamVR 后,用 PC Chrome / Edge 打开 https://127.0.0.1:5174/?preview=webxr,点击 Enter VR 进入沉浸式查看。真实模型建议通过 payload=<encoded-json> 传入 ply / poses / sceneId

桌面端查看器:

cd 3dgs_viewer/my-3dgs-viewer
npm install
npm run dev
Marker AR 模式(可选)

Marker AR 模式基于 MindAR 图像识别,将 3DGS 模型通过打印纸板锚定到真实环境中。该模式集成在 spark-3dgs-viewer 内,通过 URL 参数 mode=marker-ar 激活:

移动端 Flutter 应用中,可在模型查看页面右上角点击 AR 按钮一键切换到 Marker AR 模式,系统会自动请求摄像头权限并通过 WebView 加载 AR 场景。

使用前需准备:

  1. 生成 MindAR 目标描述文件(.mind),放入 public/targets/
  2. 打印对应的目标纸板图像
  3. 在移动端打开模型查看器,切换到 Marker AR 模式后对准纸板

5. 启动移动端 (App)

app/.env 中填入客户端所需的 Supabase 配置后运行:

cd app
flutter pub get
flutter run

当前移动端主链路已经包括:

  • 登录与会话管理
  • 视频上传与任务创建
  • 任务状态页
  • Recall 资产页
  • 基于 WebView 的移动端 WebGL 模型查看
  • Marker AR 纸板锚定模式(一键切换,自动请求摄像头权限)
  • agent-recall 驱动的 Agent 检索、多轮续聊与流式步骤面板
  • 端侧本地问答模型下载、选择与受约束问答

如需单独验证搜索 / Agent 链路,可分别在本地启动以下函数:

cd supabase/functions/search-models
supabase functions serve search-models --no-verify-jwt --env-file .env.local
cd supabase/functions/agent-recall
supabase functions serve agent-recall --no-verify-jwt

搜索接口的测试方法见 tests/README.md

数据流与存储约定

Storage 分为两个主要 bucket:braindance-assets(原始素材与重建输出)和 braindance-models(端侧模型发布)。数据库核心表包括 processing_tasksmodel_assetsmemory_posescommunity_postsworker_nodes 等。

详细的路径约定、表结构与字段说明见 supabase/README.md

语义搜索与 Agent 编排

当前仓库已包含 supabase/functions/search-models,用于承载自然语言搜索接口。它主要负责:

  1. 解析自然语言中的检索目标与时间条件。
  2. 调用 Embedding 接口生成向量。
  3. 通过 pgvector 与数据库函数检索相关场景和空间锚点。

在此基础上,supabase/functions/agent-recall 已进一步把检索能力升级为统一 Agent 入口:

  1. 统一路由 spatial_searchasset_metadatatime_comparecreativememory_graph 多种模式。
  2. 输出稳定结构 answer + evidence + actions + top_candidates + tool_trace
  3. 支持 selectedModelIdsexecutionModesessionStateconversationSummary 等上下文参数。
  4. 支持 SSE / NDJSON 流式协议,便于 Flutter 展示过程态与最终答案。
  5. 通过 follow_upsession_state 支持多轮续聊、候选确认和写操作预览重放。

本地运行示例:

cd supabase/functions/search-models
supabase functions serve search-models --no-verify-jwt --env-file .env.local

文档导航 (Documentation)

如果你是第一次接触这个项目,建议按下面的顺序阅读:

参考与致谢 (Acknowledgements)

BrainDance 的三维重建与语义理解能力建立在多个优秀开源项目和研究工作的基础上。这里保留项目级简版致谢,详细论文引用、许可证与上游说明见 ai_engine/3dgs/THIRD_PARTY_ATTRIBUTIONS.md

核心算法与三维重建

  • Nerfstudio - 模块化 NeRF / 3DGS 训练框架,本项目训练管线基于 splatfacto 模型修改
  • gsplat - 超快 CUDA 光栅化后端,为云端训练提供性能保障
  • 3D Gaussian Splatting (Inria) - Inria 原始论文实现,奠定 3DGS 理论基础
  • 2D Gaussian Splatting - 2DGS 表达与几何重建
  • Sparse2DGS - 稀疏视角下的 2DGS 重建
  • SuGaR - 面向网格提取与精修的 Gaussian 表达
  • Depth Anything 3 - 深度估计与视图空间恢复
  • SHARP - Apple 单图快速视图合成与 Gaussian 生成
  • SAM 3D Objects - 单图 / 少图 3D 物体生成
  • MoGe (Microsoft) - 单目几何估计,SAM 3D Objects 的深度后端
  • COLMAP - 结构化运动恢复 (SfM),Nerfstudio 训练管线的前置依赖
  • GLOMAP - 全局 SfM 重建
  • PyTorch - 深度学习框架,AI 引擎的基础
  • Ultralytics YOLO - 目标检测与图像分割

Web 渲染与查看器

基础设施与 AI 服务

  • Supabase - Auth、Storage、Realtime 与 Edge Functions 基础设施
  • pgvector - PostgreSQL 向量扩展,为项目提供高性能 RAG 检索能力
  • LangChain - Agent 编排框架,项目后端智能路由与工具调用的核心
  • OpenAI API - LLM API 服务
  • Qwen-VL - 通义千问多模态大模型,为 3D 场景提供自动标签与描述能力
  • DashScope (阿里云) - 通义千问模型服务
  • Deno - Supabase Edge Functions 运行时

移动端与前端

版权与开源协议 (License & Copyright)

BrainDance (流光·记) 的所有核心工程架构与业务代码的知识产权均归属原作者及项目贡献者所有。

本项目采用 PolyForm Noncommercial License 1.0.0 协议对外公开源码。

  • 允许:个人学习、学术研究、非营利组织使用以及二次修改
  • 严禁任何商业用途:包括但不限于企业内部生产使用、收费服务、闭源商业集成等

第三方依赖特别声明 本项目 AI 引擎部分集成了多个上游研究项目与子模块,使用时需同时遵守其各自许可证与非商业研究协议。

About

🧠 “Ctrl+F” for Reality. 并不是我们在回忆过去,而是过去依然驻留在那里。BrainDance 利用 3DGS 技术,将瞬间凝固为可交互、可回溯的全息时空。

Topics

Resources

License

Stars

Watchers

Forks

Packages

 
 
 

Contributors