This commit is contained in:
LinZhuoChen
2026-04-18 02:26:23 +08:00
parent 3d1fe1d8e2
commit 01c99afc41
329 changed files with 34 additions and 7 deletions

View File

@@ -104,11 +104,11 @@ python demo.py --model_path /path/to/checkpoint.pt \
# University scene
python demo.py --model_path /path/to/checkpoint.pt \
--image_folder example/university4 --mask_sky
--image_folder example/university --mask_sky
# Loop scene (loop closure trajectory)
python demo.py --model_path /path/to/checkpoint.pt \
--image_folder example/loop --mask_sky
--image_folder example/loop
```
### Streaming Inference from Images
@@ -197,6 +197,17 @@ If you run into out-of-memory issues, try one (or both) of the following:
- **`--offload_to_cpu`** — offload per-frame predictions to CPU during inference (on by default; use `--no-offload_to_cpu` only if you have memory to spare).
- **`--num_scale_frames 2`** — reduce the number of bidirectional scale frames from the default 8 down to 2, which shrinks the activation peak of the initial scale phase.
### Faster Inference
Lower the number of iterative refinement steps in the camera head to trade a small amount of pose accuracy for wall-clock speed:
```bash
python demo.py --model_path /path/to/checkpoint.pt \
--image_folder /path/to/images/ --camera_num_iterations 1
```
`--camera_num_iterations` defaults to `4`; setting it to `1` skips three refinement passes in the camera head (and shrinks its KV cache by 4×).
# 📜 License
This project is released under the Apache License 2.0. See [LICENSE](LICENSE.txt) file for details.