Given groups=1, weight of size [3, 6, 1, 1], expected input[3, 4, 256, 256] to have 6 channels, but got 4 channels instead

hellloxiaotian / DudeNet

Designing and Training of A Dual CNN for Image Denoising (Knowledge-based Systems, 2021)

https://arxiv.org/pdf/2007.03951.pdf

47 stars 5 forks source link

Given groups=1, weight of size [3, 6, 1, 1], expected input[3, 4, 256, 256] to have 6 channels, but got 4 channels instead #2

Open saimayasmeen4 opened 3 years ago

saimayasmeen4 commented 3 years ago

Hi, in dudenet you have done experiments for real noisy images also. However, when I provide grayscale images to model it shows below error RuntimeError: Given groups=1, weight of size [3, 6, 1, 1], expected input[3, 4, 256, 256] to have 6 channels, but got 4 channels instead

so can you please let me know why this error occur or what need to be changed in the code