LiteVGGT is a 3D vision foundation model that significantly boosts vanilla VGGT's performance by achieving up to 10x speedup and substantial memory reduction. This enables efficient processing of large-scale scenes (up to 1000 images) for 3D reconstruction, while maintaining high accuracy in camera pose and point cloud prediction. The method introduces a geometry-aware cached token merging strategy to optimize anchor token selection and reuse merge indices, preserving key geometric information with minimal accuracy impact.
First, create a virtual environment using Conda, clone this repository to your local machine, and install the required dependencies.
1conda create -n litevggt python=3.10
2conda activate litevggt
3git clone git@github.com:GarlicBa/LiteVGGT-repo.git
4cd LiteVGGT-repo
5pip install -r requirements.txt
Install the Transformer Engine package following its official installation requirements (see
https://github.com/NVIDIA/TransformerEngine):
1export CC=your/gcc/path
2export CXX=your/g++/path
3pip install --no-build-isolation transformer_engine[pytorch]
1python run_demo.py \
2 --ckpt_path path/to/your/te_dict.pt \
3 --img_dir path/to/your/img_dir \
4 --output_dir ./recon_result \