I’m currently working on Image denoising task.
I have two folders with me- one folder contains clean images and the other folder contains noisy images.
NOTE: the images in both the folders are in sequence. i.e. If clean folder has an image 1.png then 1.png is the noisy image corresponding to that clean image.
I’ve selected MSE loss as my loss class
Now the problem I’m facing is that when I’m writing a loop to train my CNN, I have to feed both the output of the model ( output = model(noisy) ) and the clean image
I don’t know how to load both the images simultaneously in the training loop.
Can anyone help me out with it.
Thanks in advance!