search

wanmeihuali / taichi_3d_gaussian_splatting

An unofficial implementation of paper 3D Gaussian Splatting for Real-Time Radiance Field Rendering by taichi lang.
Apache License 2.0
670 stars 62 forks source link

Speeding up training stage #130

Open hardikdava opened 1 year ago

hardikdava commented 1 year ago

I am new to gaussian splatting and taichi as well. Is there quick way to speed up training process without loosing quality. Is it possible to do batch processing?

P.S. The training process takes more than 1 hour for 20k iterations on Tesla T4.

wanmeihuali commented 1 year ago

I think so. It's in previous TODO list. But I don't have that much time to implement it... It shall not be difficult, just need to add a new dim in sorting key(tile uv to duv), and also add the dim in rasterization/backward. As we only use less than 4GB GPU memory now, I think it shall help.

jb-ye commented 1 year ago

@wanmeihuali do you still have plans for batch processing?

wanmeihuali commented 1 year ago

@Alexma3312 @Alexma3312 @yanzhoupan Do you have time to take a look?

yanzhoupan commented 1 year ago

Will start working on batch training.