Instructions to use zimhe/SpatialDiffusion with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use zimhe/SpatialDiffusion with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("zimhe/SpatialDiffusion", torch_dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
| # π Spatial Diffusion | |
| **Spatial Diffusion** is a generative model for synthesizing **spatial panoramas** based on a **cubemap representation**. By generating six orthogonal cube faces (front, back, left, right, top, bottom), the model constructs a complete and spatially consistent 360Β° view of a scene. This cubemap-based approach ensures geometric coherence and enables immersive scene generation for various downstream applications. | |
| ## π Model Highlights | |
| - **Cubemap Representation** | |
| Generates six cube faces to represent the entire spherical environment, maintaining consistent spatial alignment. | |
| - **Diffusion-Based Generation** | |
| Uses a diffusion process to progressively refine spatial details and structure, producing high-quality and coherent outputs. | |
| - **360Β° View Synthesis** | |
| Capable of producing panoramas suitable for virtual reality, robotics, and simulation environments. | |
| ## π Intended Applications | |
| - Virtual Reality (VR) scene generation | |
| - Environmental simulation and reconstruction | |
| - Robotics & autonomous navigation (spatial awareness) | |
| ## β οΈ Limitations | |
| - Performance may drop in scenes with non-Euclidean geometry or extreme occlusions. | |
| - Post-processing may be required for equirectangular projection if not viewed via cubemap renderers. | |
| - May not generalize well outside the distribution of the training dataset. | |
| ## π Citation | |
| If you use this model in your research or application, please cite: | |
| Spatial Diffusion: Cubemap-Based Generation of Spatial Panoramas, [Ziming He], 2025. | |