cuiziteng / Aleth-NeRF

🌕 [AAAI 2024] Aleth-NeRF: Illumination Adaptive NeRF with Concealing Field Assumption (Low-light enhance / Exposure correction + NeRF)
Apache License 2.0
76 stars 5 forks source link

How to adjust the enhance degree and contrast degree for over exposure dataset #13

Closed lastbasket closed 6 months ago

lastbasket commented 6 months ago

Hi, I am training on my own over-exposure dataset. I find that the enhance degree and contrast degree are difficult to adjust and will result in nan in terms of training loss and psnr.

cuiziteng commented 6 months ago

Hi, thanks for using our project ~

Due to I can't get your own data, so I guess that there maybe some problem occurred in the loss function part. Please check the value in the inverse tone mapping loss function, if this is problem, please change it to the original MSE loss function.

https://github.com/cuiziteng/Aleth-NeRF/blob/7e1ef4116552faedb1af3966fb41e79a61eb0f2a/src/model/aleth_nerf_exp/helper.py#L23

lastbasket commented 6 months ago

Thanks, I clip both x, y and the problem is resolved.