Create demo_nerfstudio.py

[najmemhmdb]

Pi3 NeRF Studio Export Demo

Overview

demo_nerfstudio.py is a comprehensive demo script that runs Pi3 model inference on image sequences and exports the results in NeRF Studio format. This script bridges the gap between Pi3's 3D reconstruction capabilities and NeRF-based novel view synthesis workflows.

Features

• Pi3 Model Inference: Runs the Pi3 model on image sequences or videos
• Camera Intrinsics Estimation: Automatically estimates camera parameters from 3D point clouds
• NeRF Studio Export: Exports camera poses and intrinsics in NeRF Studio JSON format
• Point Cloud Export: Optional export of colored point clouds in PCD format
• Image Export: Saves processed images for NeRF training

Usage

Basic Usage

python demo_nerfstudio.py --data_path /path/to/images --save_path /path/to/output

Output Structure

The script creates the following output structure:
output_directory/
├── transforms.json # NeRF Studio camera poses and intrinsics
├── camera/ # Exported images
│ ├── 000.jpg
│ ├── 001.jpg
│ └── ...
└── pcd/ # Point clouds (if --save_points enabled)
├── 000.pcd
├── 001.pcd
└── ...

TODO

*[ ] Revert images and intrinsic to original size
*[ ] Save depth images to train Depth-NeRFacto