Skill-3D
收藏资源简介:
Skill-3D是一个用于代理3D空间推理的框架,其数据集部分包含四个3D空间推理基准测试:VSI-Bench、BLINK、CV-3D/CV-Bench和MMSI-Bench。数据集布局包括skill3d_splits文件夹,其中包含训练和测试JSONL文件,用于技能演化、后训练和评估。数据集通过Hugging Face发布,仅从训练样本中收集教师生成的rollouts。
Skill-3D is a framework for agent 3D spatial reasoning. Its dataset component includes four 3D spatial reasoning benchmarks: VSI-Bench, BLINK, CV-3D/CV-Bench, and MMSI-Bench. The dataset layout consists of the skill3d_splits folder, which contains training and test JSONL files for skill evolution, post-training, and evaluation. The dataset is released via Hugging Face, with only teacher-generated rollouts collected from training samples utilized.
数据集概述:Skill-3D
Skill-3D 是一个用于智能体3D空间推理(Agentic 3D Spatial Reasoning)的框架与数据集。其核心创新在于通过构建“场景记忆”(Scene Memory)和演化“技能库”(Skill Library),使多模态大语言模型(MLLM)智能体能够针对不同3D场景自适应地选择和使用外部感知与几何工具,从而提升空间推理任务的性能。
关键特性
- 场景感知技能提取:成功使用工具的路径被提炼为可复用的工作流(技能),失败的路径被保留为纠正性教训。
- 场景记忆与技能库:从所有基准测试的训练集样本中构建共享的记忆与技能库,确保测试样本不被用于技能构建或后训练。
- 显著提升工具利用率:在VSI-Bench基准上,有效工具使用率从39%提升至78%。
数据集构成与评估基准
Skill-3D在以下四个3D空间推理基准上进行评估:
| 基准名称 | 主要评估任务 |
|---|---|
| VSI-Bench | 物体计数、公制距离、物体大小、房间大小、相对方向、路线规划、外观顺序 |
| BLINK | 多视图空间推理 |
| CV-3D / CV-Bench | 深度顺序和相对距离推理 |
| MMSI-Bench | 位置关系推理 |
数据布局与发布
- 数据集在Hugging Face上发布,包含用于技能演化、后训练和评估的
skill3d_splits文件。 - 典型的本地数据布局如下(
dataset/为根目录):
text
dataset/
|-- skill3d_splits/
| |-- vsi_train.jsonl
| |-- vsi_test.jsonl
| |-- mmsi_pr_train.jsonl
| |-- mmsi_pr_test.jsonl
| |-- cv3d_train.jsonl
| |-- cv3d_test.jsonl
| |-- blink_multiview_train.jsonl
| -- blink_multiview_test.jsonl |-- VSI-Bench/ |-- MMSI-Bench/ |-- CV-3D/ -- BLINK/
工具服务与环境
Skill-3D集成了多个外部专家工具,均需独立部署服务。典型工具及其环境变量和默认端口如下:
| 工具名称 | 环境变量 | 默认URL | 功能 |
|---|---|---|---|
| Depth-Anything-3 | DEPTH_SERVER_URL |
http://127.0.0.1:20019 |
公制深度估计 |
| SAM3 | SEGMENTATION_SERVER_URL |
http://127.0.0.1:20020 |
图像分割 |
| GroundingDINO | DETECTION_SERVER_URL |
http://127.0.0.1:20022 |
开放词汇目标检测 |
| Pi3 | PI3_SERVER_URL |
http://127.0.0.1:20030 |
3D重建 |
| SwinIR | SWINIR_SERVER_URL |
http://127.0.0.1:20032 |
超分辨率/图像恢复 |
| Orient-Anything v2 | ORIENT_ANYTHING_SERVER_URL |
http://127.0.0.1:20034 |
物体方向估计 |
后训练与推理
- 训练:提供SFT(监督微调)和GRPO(基于组相对策略优化)的Shell脚本和提示模板。已训练好的4B/8B参数检查点已在Hugging Face上发布。
- 推理:支持通过OpenAI兼容的API或本地vLLM服务器进行推理。
仓库结构
项目核心代码位于 skill3d/ 目录下,包括智能体、记忆、技能、工具注册和模型封装等模块。
许可证
该项目采用 Apache-2.0 许可证。




