Validation in training loop - dimensionality problems about pd-denoising-pytorch HOT 2 OPEN

Hi Tom, the codes are very old so I am not sure whether my answer will address your problem.
Basically, in the paper we used two noise types: AWGN and RVIN, so the noise map contains 6 channels (RGB for AWGN and RGB for RVIN).

I guess the commented validation codes are deprecated and it only generates one single type AWGN noise levels. It's better to refer https://github.com/yzhouas/PD-Denoising-pytorch/blob/master/train.py#L111-L133 to generate the validation noise level maps.

The codes are not clean enough and sorry for causing so many troubles.

from pd-denoising-pytorch.

Hi Yuqian,

first of all thanks for the quick reply!

I replaced the deprecated commands with newer versions.
In the meantime I just replicated the 3 channel generated noise map twice so it fits the 6 channels, and the code runs now. I'll review the lines you pointed to and try to adjust.

BTW, I see that "parepare_data" can create a validation data loader as well.
I guess that if I want to use it for validation instead of just running on files in the val folder I'll have to write the validation loop myself, right?

Thanks again,
Tom

from pd-denoising-pytorch.