VAST-AI-Research/TriplaneGaussian

ckpt cannot set use_rgb == true

YuanSnowing opened this issue · 2 comments

Thanks a lot for your perfect work!
I want to use your model to provide an initialized 3dgs, so I tried to set use_rgb == true. But this will lead the following error:
image
Could you upload an RGB-type model ? Or is there an approach to transfer an shs-3dgs to rgb-3dgs?

BTW, the colab may add one line to prevent error:
!pip install "jax[cuda12_local]==0.4.23" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

We haven't trained the model with rgb-3dgs. If you prefer to use only rgb, perhaps you can manually compute it from shs?
Thank you for the reminder, we will fix it later!