pytorch implementation of NUDCP Paper.
Primarily used for predicting of the medium transmission map in underwater environments.
The code is translated from the original python version from Github repository.
Tensor operations are changed from NumPy to Torch and can be applied in cuda.
@article{Underwater Image Enhancement Method,
author = {Wei Song, Yan Wang, Dongmei Huang, Antonio Liotta, Cristian Perra},
title = {Enhancement of Underwater Images with Statistical Model of Background Light and Optimization of Transmission Map},
journal = {IEEE Transactions on Broadcasting},
year = {2019}
}