RuntimeError: Expected 4-dimensional input for 4-dimensional weight 3 3 1, but got 3-dimensional inp

源代码GitHub - sanghyun-son/EDSR-PyTorch: PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017) - GitHub - sanghyun-son/EDSR-PyTorch: PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)https://github.com/sanghyun-son/EDSR-PyTorch在我使用原始权重EDSRx2.pt，测试DIV2K（801-900）时，报错：

Traceback (most recent call last): File "D:/EDSR-PyTorch-master/src/main.py", line 35, in <module> main() File "D:/EDSR-PyTorch-master/src/main.py", line 28, in main while not t.terminate(): File "D:\EDSR-PyTorch-master\src\trainer.py", line 160, in terminate self.test() File "D:\EDSR-PyTorch-master\src\trainer.py", line 109, in test sr = self.model(lr, idx_scale) File "D:\Anaconda3\envs\python3.6SRDenseNet\lib\site-packages\torch\nn\modules\module.py", line 493, in __call__ result = self.forward(*input, **kwargs) File "D:\EDSR-PyTorch-master\src\model\__init__.py", line 57, in forward return forward_function(x) File "D:\EDSR-PyTorch-master\src\model\__init__.py", line 135, in forward_chop y = self.forward_chop(*p, shave=shave, min_size=min_size) File "D:\EDSR-PyTorch-master\src\model\__init__.py", line 126, in forward_chop y = P.data_parallel(self.model, *x, range(n_GPUs)) File "D:\Anaconda3\envs\python3.6SRDenseNet\lib\site-packages\torch\nn\parallel\data_parallel.py", line 204, in data_parallel return module(*inputs[0], **module_kwargs[0]) File "D:\Anaconda3\envs\python3.6SRDenseNet\lib\site-packages\torch\nn\modules\module.py", line 493, in __call__ result = self.forward(*input, **kwargs) File "D:\EDSR-PyTorch-master\src\model\edsr.py", line 56, in forward x = self.sub_mean(x) File "D:\Anaconda3\envs\python3.6SRDenseNet\lib\site-packages\torch\nn\modules\module.py", line 493, in __call__ result = self.forward(*input, **kwargs) File "D:\Anaconda3\envs\python3.6SRDenseNet\lib\site-packages\torch\nn\modules\conv.py", line 338, in forward self.padding, self.dilation, self.groups) RuntimeError: Expected 4-dimensional input for 4-dimensional weight 3 3 1, but got 3-dimensional input of size [1, 184, 270] instead

 原因是维度问题，增加一个维度即可，

在 Model/init.py中的forward_chop()函数进行修改，

修改为

  for p in zip(*x_chops):
                
                p1 = [p[0].unsqueeze(0)]
         
                y = self.forward_chop(*p1, shave=shave, min_size=min_size)
                if not isinstance(y, list): y = [y]
                if not y_chops:
                    y_chops = [[_y] for _y in y]
                else:
                    for y_chop, _y in zip(y_chops, y): y_chop.append(_y)

 参考链接：It seems that there are some problems with --chop argument · Issue #184 · sanghyun-son/EDSR-PyTorch · GitHubhttps://github.com/sanghyun-son/EDSR-PyTorch/issues/184