UWCNN
UWCNN copied to clipboard
Code and Datasets for "Underwater Scene Prior Inspired Deep Underwater Image and Video Enhancement", Pattern Recognition, 2019
> Choose the needed model (For example, if you need the type 1 model, please put 'model_checkpoint_path: "coarse.model-type1"' on the first line of checkpoint text.If you are using existing checkpoint...
Hi, Can you please tell which part of the code describes about the different loss incurred in the architecture(UWCNN)? That is MSE, PNSR and SSIM? Thanks,
Ex,I want to know how your model is trained, it seems that there is no interface to put the training set for training this code.I really need your answer,please.
UWCNN+
Hey @saeed-anwar , I was looking around and i noticed that there is nothing about the implementation for UWCNN+. I was hopping to work on it if it's not available...
Ex me, we have some troubles in training our model. We can't find the definition of your loss function. So we are disable to understand the processing of model training....
Fixed #2 Which will return segmentation error while using high resolution images Updated utils.py to resize the images which will be the input to the height and width specified in...
segmentation fault occurs when using high resolution images while running main_test.py