DirectVoxGO
DirectVoxGO copied to clipboard
fix-total_variation_kernel_bug
Thank for the nice work ! In function of __global__ void total_variation_add_grad_cuda_kernel in the file total_variation_kenel.cu, I found that there may be a problem in add grad.
when reading the paper, I think that this function means to deal with the Aliasing in the voxel grid and let the voxel grid's value become more smooth.Therefore, if in 3D voxel grid ,the method of add variation in voxel grid maybe is to add grad in axis x,y,z. But I found that axis z was reused, so I change it to 'wx' in gradient axis 'x'.
I also run the experiment between the changes ,I found that the PSNR will raise by 0.02 in llff dataset ,Scene room.