Wenchao-Du / LIR-for-Unsupervised-IR

This is an implementation for the CVPR2020 paper "Learning Invariant Representation for Unsupervised Image Restoration"
https://arxiv.org/pdf/2003.12769.pdf
98 stars 21 forks source link

学长您好,我在构建自己数据集进行训练的时候发生错误,尝试一天后没找到解决办法,真诚希望学长能提点一下 #16

Open Furxier opened 9 months ago

Furxier commented 9 months ago

我的数据集是640*512的红外噪声图像(Celeba_B)与干净图像(Celeba_A),但是一直在 File "G:\WIN\LIR-for-Unsupervised-IR-master\networks.py", line 344, in forward return self.model(x) 与 File "G:\WIN\LIR-for-Unsupervised-IR-master\networks.py", line 467, in forward x = self.conv(self.pad(x))的 def forward(self, x): x = self.conv(self.pad(x)) if self.norm_type != 'wn' and self.norm != None: x = self.norm(x) if self.activation: x = self.activation(x) return x 报错 return F.pad(input, self.padding, 'reflect') TypeError: pad(): argument 'input' (position 1) must be Tensor, not NoneType 实在毫无头绪,恳求学长帮助