Skip to content

Repository files navigation

OpenSpatial Logo

arXiv Hugging Face License: Apache-2.0

OpenSpatial is an open-source 3D spatial understanding data engine engineered for high quality, extensive scalability, broad task diversity, and optimized efficiency.

By bridging the gap between massive 2D web data and complex 3D spatial reasoning, OpenSpatial provides a comprehensive suite for the next generation of Embodied AI and World Models.


OpenSpatial Teaser
OpenSpatial Pipeline: From 2D Web Data to 3D Spatial Understanding


🔥 News

  • [2026.04.15] 🎉 We have released the open-source subset of the OpenSpatial-3M dataset! Check it out on Hugging Face.
  • [2026.04.08] 🎉 The OpenSpatial 3D data engine is now officially open-sourced.

🚀 Key Features

  • Web Data 3D Lifting: Advanced pipelines to transform large-scale 2D web imagery into geometrically consistent 3D representations.
  • Diverse Data Generation: Automated engine for creating rich spatial understanding datasets, covering various environments and object-level details.
  • Multi-Task Integration: Support for a wide range of tasks including 3D grounding, spatial reasoning, and scene captioning.
  • Comprehensive Evaluation: Built-in benchmarking suite to evaluate spatial understanding capabilities across different model architectures.
  • High Efficiency: Optimized for large-scale data processing with scalable distributed computing support.

📊 Dataset

The OpenSpatial-3M dataset is now available on Hugging Face. It contains 3 million high-fidelity samples designed to enhance 3D spatial reasoning in large multi-modal models.

📖 Documentation

Document Description
Quick Start Data preparation, config structure, annotation pipeline usage, and running tasks end-to-end
Development Guide Adding new annotation tasks, pipeline stages, prompt templates, dataset preprocessors, and internal architecture reference

🗺️ Cognitive Map & Benchmark-Style Tasks

OpenSpatial optionally attaches a question-related cognitive map to every generated QA. A cognitive map is a 10×10 bird's-eye-view grid recording the cameras (position + yaw) and objects (position + size + yaw) that participate in the QA, plus a PNG visualization with the question/answer overlay.

Enable it per task by adding the following block to any config/annotation/*.yaml:

cognitive_map:
  enable: true               # turn the feature on (off by default for back-compat)
  enable_visualization: true # render PNGs alongside the JSON map
  dump_samples: true         # dump the first N PNGs to output_dir/cognitive_map_samples/
  dump_sample_count: 20
  grid_size: 10
  padding_ratio: 0.10

The following benchmark-style multiview tasks are available out of the box (see their dedicated demo YAMLs under config/annotation/):

Task Benchmark style YAML
Relative Distance all-angles demo_multiview_relative_distance.yaml
Clockwise (Yes/No) BLINK demo_multiview_clockwise.yaml
Camera Movement Direction VSI-Bench demo_multiview_camera_movement.yaml
Camera–Camera MMSI-Bench demo_mmsi_camera_camera.yaml
Camera–Object MMSI-Bench demo_mmsi_camera_object.yaml
Camera–FacingObject–Camera MMSI-Bench (extended) demo_mmsi_camera_facing_object_camera.yaml
Object–Object (world-frame) MMSI-Bench demo_mmsi_object_object.yaml
Object–FacingObject–Object MMSI-Bench (extended) demo_mmsi_object_facing_object_object.yaml
Camera Motion (composite) MMSI-Bench demo_mmsi_camera_motion.yaml
BEV Pose Estimation all-angles demo_multiview_bev_pose_estimation.yaml
Manipulation Viewpoint all-angles (approx.) demo_multiview_manipulation_view.yaml

📅 Roadmap & To-Do List

  • 3D Data Engine: Open-source the core 3D spatial understanding data engine.
  • OpenSpatial-3M Dataset Release: Publicly release the large-scale 3M spatial understanding dataset. [HF Link]
  • Model Release: Release the trained spatial understanding model.
  • Evaluation Suite: Open-source the comprehensive evaluation code for spatial tasks.
  • 3D Lifting Module: Integrate the core engine for lifting 2D web data to 3D representations.
  • More Tasks: Extend support for more spatial understanding task types.

📄 Citation

If you find OpenSpatial useful for your research, please consider citing our paper:

@article{openspatial2025,
  title={OpenSpatial: An Open-Source 3D Spatial Understanding Data Engine},
  journal={arXiv preprint arXiv:2604.07296},
  year={2025}
}

About

No description, website, or topics provided.

Resources

Stars

2 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages