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

Update the implementation of ssim() loss function, reduce the computational complexity from O(n) to O(1) since create_window() is called only once in main function

Open tryhiseyyysum opened this issue 1 year ago • 0 comments

In the loss_utils.py: Updated the function ssim(), the optimized version of ssim() is ssim_optimized(), which reduce the computational complexity. After modified, the function create_window() just need to be called once in the main function in train.py, no need to be called in ssim() in every iteration.

In the train.py: Updated the main function, call the create_window() function for just once, since the window_size is fixed to 11 and the channel is fixed to 3 according to the source code. Updated the calculation of loss function, call the ssim_optimized(), instead of the original ssim().

tryhiseyyysum avatar Jul 12 '24 09:07 tryhiseyyysum