SceneGen: Single-Image 3D Scene Generation in One Feedforward Pass (3DV 2026)

This repository contains the official PyTorch implementation of SceneGen, introduced in SceneGen: Single-Image 3D Scene Generation in One Feedforward Pass.

Now the Training, Inference Code, and Pretrained Models have all been released! Feel free to reach out for discussions!

🌟 Resources

Project Page · Paper · Code · Checkpoints

⏩ News

• [2025.11] Evaluation code has been released.
• [2025.11] Glad to share that SceneGen has been accepted to 3DV 2026.
• [2025.9] Our training code and data processing code are released.
• [2025.8] The inference code and checkpoints are released.
• [2025.8] Our pre-print paper has been released on arXiv.

📦 Installation & Pretrained Models

Prerequisites

• Hardware: An NVIDIA GPU with at least 16GB of memory is necessary. The code has been verified on NVIDIA A100 and RTX 3090 GPUs.
• Software:
  • The CUDA Toolkit is needed to compile certain submodules. The code has been tested with CUDA versions 12.1.
  • Python version 3.8 or higher is required.

Installation Steps

1. Clone the repo:

   git clone https://github.com/Mengmouxu/SceneGen.git
   cd SceneGen
   

2. Install the dependencies: Create a new conda environment named scenegen and install the dependencies:

   . ./setup.sh --new-env --basic --xformers --flash-attn --diffoctreerast --spconv --mipgaussian --kaolin --nvdiffrast --demo
   

   The detailed usage of setup.sh can be found by running . ./setup.sh --help.

Pretrained Models

1. First, create a directory in the SceneGen folder to store the checkpoints:

   mkdir -p checkpoints
   

2. Download the pretrained models for SAM2-Hiera-Large and VGGT-1B from SAM2 and VGGT, then place them in the checkpoints directory. (SAM2 installation and its checkpoints are required for interactive generation with segmentation.)
3. Download our pretrained SceneGen model from here and place it in the checkpoints directory as follows:

   SceneGen/
   ├── checkpoints/
   │   ├── sam2-hiera-large
   │   ├── VGGT-1B
   │   └── scenegen
   |       ├──ckpts
   |       └──pipeline.json
   └── ...
   

💡 Inference

We provide two scripts for inference: inference.py for batch processing and interactive_demo.py for an interactive Gradio demo.

Interactive Demo

This script launches a Gradio web interface for interactive scene generation.

• Features: It uses SAM2 for interactive image segmentation, allows for adjusting various generation parameters, and supports scene generation from single or multiple images.
• Usage:

  python interactive_demo.py
  

  🚀 Quick Start Guide

  📷 Step 1: Input & Segment

  1. Upload your scene image.
  2. Use the mouse to draw bounding boxes around objects.
  3. Click "Run Segmentation" to segment objects.

  ※ For multi-image generation: maintain consistent object annotation order across all images.

  🗃️ Step 2: Manage Cache

  1. Click "Add to Cache" when satisfied with the segmentation.
  2. Repeat Steps 1-2 for multiple images.
  3. Use "Delete Selected" or "Clear All" to manage cached images.

  🎮 Step 3: Generate Scene

  1. Adjust generation parameters (optional).
  2. Click "Generate 3D Scene".
  3. Download the generated GLB file when ready.

Watch the demo video

Pre-segmented Image Inference

This script processes a directory of pre-segmented images.

• Input: The input folder structure should be similar to assets/masked_image_test, containing segmented scene images.
• Visualization: For scenes with ground truth data, you can use the --gradio flag to launch a Gradio interface that visualizes both the ground truth and the generated model.
• Usage:

  python inference.py --gradio
  

📚 Dataset

To train and evaluate SceneGen, we use the 3D-FUTURE dataset. Please refer to the GitHub repository for detailed preprocessing and data handling instructions.

🏋️‍♂️ Training

With the processed 3D-FUTURE dataset and the pretrained ss_flow_img_dit_L_16l8_fp16.safetensors model checkpoint from TRELLIS correctly placed in the checkpoints/scenegen/ckpts directory, you can train SceneGen using the following command:

bash scripts/train.sh

🧪 Evaluation

To generate the 3D scenes on the 3D-FUTURE test set using the SceneGen model, use the following command:

bash scenegen_eval.sh

which will use the scenegen_eval.py script to generate the normalized scenes.

To evaluate the trained SceneGen model on the 3D-FUTURE test set, use the following command:

cd evalscene
bash eval_scenegen.sh

Make sure to have the processed 3D-FUTURE dataset and the rendered images in place as described in the Dataset section and the evaluation configs in evalscene/configs/test/scene_evaluation_scenegen.yaml set correctly. Then the evaluation script will compute metrics between the normalized generated scenes and the ground truth.

Some packages used in the evaluation require additional installation. Please install the packages: torchmetrics, lpips, clip, and probreg via pip.

📜 Citation

If you use this code and data for your research or project, please cite:

@inproceedings{meng2026scenegen,
  author    = {Meng, Yanxu and Wu, Haoning and Zhang, Ya and Xie, Weidi},
  title     = {SceneGen: Single-Image 3D Scene Generation in One Feedforward Pass},
  booktitle   = {International Conference on 3D Vision 2026},
  year      = {2026},
}

Acknowledgements

Many thanks to the code bases from TRELLIS, DINOv2, and VGGT.

Contact

If you have any questions, please feel free to contact [email protected] and [email protected].