DekuLiuTesla / CityGaussian

CityGaussian Series for High-quality Large-Scale Scene Reconstruction with Gaussians
https://dekuliutesla.github.io/CityGaussianV2
Other
450 stars 29 forks source link

Lod rendering error #24

Closed WQ99852 closed 2 months ago

WQ99852 commented 2 months ago

Hello, I trained the model with my custom data, the normal rendering is work, and LoD rendering occurred the following error:

Loading trained model at iteration 30000 [04/09 17:23:10] Reading camera 432/432 [04/09 17:23:13] Train cameras: 432, Test cameras: 0 [04/09 17:23:13] Init LoD 3 with 20983781 points from output/test_light_40_vq [04/09 17:23:27] Traceback (most recent call last): File "/code/CityGaussian/render_large_lod.py", line 150, in render_sets(lp, args.iteration, pp, args.load_vq, args.skip_train, args.skip_test, args.custom_test) File "/code/CityGaussian/render_large_lod.py", line 102, in render_sets lod_gs = BlockedGaussian(lod_gs, lp, compute_cov3D_python=pp.compute_cov3D_python) File "/code/CityGaussian/scene/gaussian_model.py", line 443, in init self.cell_divider(gaussians) File "/code/CityGaussian/scene/gaussian_model.py", line 466, in cell_divider xyz_median = torch.median(xyz[cell_mask], dim=0)[0] IndexError: median(): Expected reduction dim 0 to have non-zero size.