运行给的models\network_fsanet.py出现shape错误

[XiaolongQian]

运行给的models\network_fsanet.py出现错误

PS C:\Users\ASUS\Desktop\code\Benchmark\FSANet> & C:/software/Anaconda3/envs/xiaolong/python.exe c:/Users/ASUS/Desktop/code/Benchmark/FSANet/models/network_fsanet.py
torch.Size([1, 31, 31])
Traceback (most recent call last):
File "c:\Users\ASUS\Desktop\code\Benchmark\FSANet\models\network_fsanet.py", line 319, in
out = net(image, kernel_set, eptional_map)
File "C:\software\Anaconda3\envs\xiaolong\lib\site-packages\torch\nn\modules\module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "c:\Users\ASUS\Desktop\code\Benchmark\FSANet\models\network_fsanet.py", line 266, in forward
kernel2_2 = self.lrelu(self.kconv2_1(kernel_set[0, ...].unsqueeze(0))).squeeze(0)
File "C:\software\Anaconda3\envs\xiaolong\lib\site-packages\torch\nn\modules\module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "C:\software\Anaconda3\envs\xiaolong\lib\site-packages\torch\nn\modules\conv.py", line 463, in forward
return self._conv_forward(input, self.weight, self.bias)
File "C:\software\Anaconda3\envs\xiaolong\lib\site-packages\torch\nn\modules\conv.py", line 459, in _conv_forward
return F.conv2d(input, weight, bias, self.stride,
RuntimeError: Given groups=1, weight of size [25, 25, 3, 3], expected input[1, 1, 31, 31] to have 25 channels, but got 1 channels instead

一个可能的原因是因为kernel_set的尺寸问题，输入进卷积前变成了[1,H,W] 无法正确卷积 能否麻烦告诉我输入的正确尺寸应该是如何？此外若能告诉我，已知整张图的PSF map，如何选择出网络所需的kernel_set 我将感激不尽~