Chrixtar / latentsplat

[ECCV 2024] Implementation of latentSplat: Autoencoding Variational Gaussians for Fast Generalizable 3D Reconstruction
https://geometric-rl.mpi-inf.mpg.de/latentsplat/
MIT License
146 stars 3 forks source link

Reproduce the result of paper #7

Closed myutility closed 4 months ago

myutility commented 5 months ago

Hi, First, thanks for the awesome work and code. I tried to reproduce the results as baseline but had some confusion:

  1. To train by myself and get the ckpt that has the same performance to your provided ckpt. Should I simply run python3 -m src.main +experiment=co3d_hydrant with one gpu and take the last checkpoint(the one at 200000 iteration)?(The default bs is 2) Or are there any more details?
  2. After I get the trained checkpoint, how can I get the statistics on the tables provided in your paper? Thanks again!
Chrixtar commented 4 months ago

Hi,

you can either train the model yourself (option 1) or use one of the provided checkpoints directly for evaluation, skipping the training (option 2). Note that after training your model, you still need to run evaluation with your obtained checkpoint! For option 1, training is explained here. For option 2, evaluation with a given checkpoint path is explained here.

What you will get in the end is a directory with reconstructed views of test scenes. For computing the statistics, I used the src/scripts/compute_metrics.py script in combination with the evaluation config files in config/evaluation. Not that in the evaluation config file, e.g. config/evaluation/co3d_hydrant_extra.yaml, you need to specify the paths to the generated image directories from evaluation (you can comment out all other baseline methods). The example for extrapolation on CO3D hydrants would look like this: python -m src.scripts.compute_metrics evaluation=co3d_hydrant_extra

Feel free to reopen the issue in case of any further questions :)

Best Chris