Views
No views yet
1conda create -n volsplat python=3.10
2conda activate volsplat
3
4pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 xformers==0.0.27.post2 --index-url https://download.pytorch.org/whl/cu121
5pip install -r requirements.txt
6
7# Install MinkowskiEngine
8# For easier installation, we made some modifications based on https://github.com/Julie-tang00/Common-envs-issues/blob/main/Cuda12-MinkowskiEngine and included it directly in our project.
9conda install -c conda-forge openblas
10pip install ninja
11cd MinkowskiEngine
12python setup.py install
13cd .../models directory. To facilitate reproduction and comparison, we also provide pretrained weights from the baseline methods trained using same input views.| Model | Download |
|---|---|
| volsplat-re10k-256x256 | download |
| pixelsplat-re10k-baseline-256x256 | download |
| mvsplat-re10k-baseline-256x256 | download |
| transplat-re10k-baseline-256x256 | download |
| depthsplat-re10k-baseline-256x256 | download |
| ggn-re10k-baseline-256x256 | download |
wget https://s3.eu-central-1.amazonaws.com/avg-projects/unimatch/pretrained/gmflow-scale1-things-e9887eda.pth -P pretrained
wget https://huggingface.co/depth-anything/Depth-Anything-V2-Base/resolve/main/depth_anything_v2_vitb.pth -P pretrained1python -m src.main +experiment=re10k \
2data_loader.train.batch_size=1 \
3'dataset.roots'='["datasets/re10k"]' \
4dataset.test_chunk_interval=10 \
5dataset.num_context_views=6 \
6trainer.max_steps=150000 \
7model.encoder.num_scales=2 \
8model.encoder.upsample_factor=2 \
9model.encoder.lowest_feature_resolution=4 \
10model.encoder.monodepth_vit_type=vitb \
11output_dir=outputs/re10k-256x256 \
12wandb.project=VolSplat \
13checkpointing.pretrained_monodepth=pretrained/pretrained_weights/depth_anything_v2_vitb.pth \
14checkpointing.pretrained_mvdepth=pretrained/pretrained_weights/gmflow-scale1-things-e9887eda.pth1python -m src.main +experiment=scannet \
2data_loader.train.batch_size=1 \
3'dataset.roots'='["datasets/scannet"]' \
4dataset.image_shape=[256,256] \
5trainer.max_steps=100000 \
6trainer.val_check_interval=0.9 \
7train.eval_model_every_n_val=40 \
8checkpointing.every_n_train_steps=2000 \
9model.encoder.num_scales=2 \
10model.encoder.upsample_factor=2 \
11model.encoder.lowest_feature_resolution=4 \
12model.encoder.monodepth_vit_type=vitb \
13output_dir=outputs/scannet-256x256 \
14wandb.project=VolSplat \
15checkpointing.pretrained_model=models/volsplat-re10k-256x256.ckpt/models.1python -m src.main +experiment=re10k \
2data_loader.train.batch_size=1 \
3'dataset.roots'='["datasets/re10k"]' \
4dataset.test_chunk_interval=10 \
5dataset/view_sampler=evaluation \
6dataset.view_sampler.num_context_views=6 \
7dataset.view_sampler.index_path=assets/re10k_evaluation/evaluation_index_re10k.json \
8trainer.max_steps=150000 \
9model.encoder.num_scales=2 \
10model.encoder.upsample_factor=2 \
11model.encoder.lowest_feature_resolution=4 \
12model.encoder.monodepth_vit_type=vitb \
13mode=test \
14test.save_video=false \
15test.save_depth_concat_img=false \
16test.save_image=false \
17test.save_gt_image=false \
18test.save_input_images=false \
19test.save_video=false \
20test.save_gaussian=false \
21checkpointing.pretrained_model=models/volsplat-re10k-256x256.ckpt \
22output_dir=outputs/volsplat-re10k-256x256-test1python -m src.main +experiment=acid \
2data_loader.train.batch_size=1 \
3'dataset.roots'='["datasets/acid"]' \
4dataset.test_chunk_interval=10 \
5dataset/view_sampler=evaluation \
6dataset.view_sampler.num_context_views=6 \
7dataset.view_sampler.index_path=assets/acid_evaluation/evaluation_index_acid.json \
8trainer.max_steps=150000 \
9model.encoder.num_scales=2 \
10model.encoder.upsample_factor=2 \
11model.encoder.lowest_feature_resolution=4 \
12model.encoder.monodepth_vit_type=vitb \
13mode=test \
14test.save_video=false \
15test.save_depth_concat_img=false \
16test.save_image=false \
17test.save_gt_image=false \
18test.save_input_images=false \
19test.save_video=false \
20test.save_gaussian=false \
21checkpointing.pretrained_model=models/volsplat-re10k-256x256.ckpt \
22output_dir=outputs/volsplat-acid-256x256-test1@article{wang2025volsplat,
2 title={VolSplat: Rethinking Feed-Forward 3D Gaussian Splatting with Voxel-Aligned Prediction},
3 author={Wang, Weijie and Chen, Yeqing and Zhang, Zeyu and Liu, Hengyu and Wang, Haoxiao and Feng, Zhiyuan and Qin, Wenkang and Chen, Feng and Zhu, Zheng and Chen, Donny Y. and Zhuang, Bohan},
4 journal={arXiv preprint arXiv:2509.19297},
5 year={2025}
6}