2d-gaussian-splatting icon indicating copy to clipboard operation
2d-gaussian-splatting copied to clipboard

MipNeRF360 Training Time

Open Yanbing-Han opened this issue 9 months ago • 5 comments

Excellent Work! I'm evaluating the MipNeRF360 dataset on one RTX4090 GPU. The average training time on the 9 scenes is 29 min (Already move grid_x grid_y to GPU) I wonder if this training time is normal or not since you didn't report the training time on your paper. Thank you!

Yanbing-Han avatar May 07 '24 15:05 Yanbing-Han

This is a great boost!! Before moving grid_x and grid_y to GPU, the average training time on my RTX3090 GPU is about 50 min.

hbb1 avatar May 07 '24 15:05 hbb1

I implemented an unofficial version of 2DGS before.
I made some modifications : the way to get the gradient of the projected 2D center , adjust parameters like prune_interval(1500), opacity_cull(0.2) The average training time now is about 20 min while maintaining the accuracy. If you are interested, please check it out. Thank you !

Yanbing-Han avatar May 07 '24 15:05 Yanbing-Han

Great work! But It seems like that the link is broken and I cannot access. I am highly appreciated if you would be able to PR or outline the changes. I will perform testing on it then. Thank you for your great contribution!

hbb1 avatar May 07 '24 16:05 hbb1

Great. I found the correct link in your github profile. I will check it later.

hbb1 avatar May 07 '24 16:05 hbb1

I implemented my version referencing your python demo, so our cuda implementation is actually quite different. I made some modifications on the official version based on my experience but it didn't improve the speed like I thought. I may need more time to check the differences.

Yanbing-Han avatar May 08 '24 01:05 Yanbing-Han