RuntimeError: Given groups=1, weight of size [1, 1, 3, 3], expected input[1, 3, 1402, 1200] to have

RuntimeError: Given groups=1, weight of size [1, 1, 3, 3], expected input[1, 3, 1402, 1200] to have 1 channels, but got 3 channels instead

通过翻译得知这是因为数据读入的通道不对，应该是1通道的，但是我们所得确是3通道的。通过查看读取的数据img的信息，发现mode是RGB。

但是我们需要的是一通道的灰度图，所以需要将mode模式改变一下。
原代码为：

o_tensor = ToTensor()
to_pil = ToPILImage()input = to_tensor(img).unsqueeze(0)
kernel = t.ones(3,3)/-9
kernel[1][1] = 1
conv = nn.Conv2d(1,1,(3,3),3,bias=False)
conv.weight.data = kernel.view(1,1,3,3)out = conv(V(input))
to_pil(out.data.squeeze(0))

错误：

RuntimeError: Given groups=1, weight of size [1, 1, 3, 3], expected input[1, 3, 1402, 1200] to have 1 channels, but got 3 channels instead

修改

o_tensor = ToTensor()
to_pil = ToPILImage()
img = Image.open('images/SDZ.jpeg')
if img.mode != 'L':img = img.convert('L')   #仅仅加上这两行就可以了input = to_tensor(img).unsqueeze(0)
kernel = t.ones(3,3)/-9
kernel[1][1] = 1
conv = nn.Conv2d(1,1,(3,3),3,bias=False)
conv.weight.data = kernel.view(1,1,3,3)out = conv(V(input))
to_pil(out.data.squeeze(0))

完成！解决问题

解决链接来源：
https://www.codeleading.com/article/31383072717/

RuntimeError: Given groups=1, weight of size [1, 1, 3, 3], expected input[1, 3, 1402, 1200] to have相关推荐

(36)RuntimeError: Given groups=4, weight of size [4, 1, 11, 11], expected input xxxxxxxxx
问题描述: 在测试AODNet去雾网络时,计算测试集的指标ssim和psnr,从test_loader中读取测试集的清晰图像和去雾之后的图像,作为ssim的输入进行计算,原文代码如下: for ite ...
RuntimeError: Given groups=1, weight of size [14, 464, 1, 1], expected input问题解决
BUG解决:RuntimeError: Given groups=1, weight of size [14, 464, 1, 1], expected input[16, 116, 56, 1] t ...
Pytorch RuntimeERROR: Given groups=1 weights of size [256,64,1,1] expected input[1,16,256,256] to
错误 Pytorch RuntimeERROR: Given groups=1 weights of size [256,64,1,1] expected input[1,16,256,256] to ...
BUG解决：RuntimeError:Given groups=1,weight of size...expected input...but got 3 channels instead.
https://www.codeleading.com/article/31383072717/
[pcl::VoxelGrid::applyFilter] Leaf size is too small for the input dataset. Integer indices would ov
1. 报错日志: Python-pcl 点云下采样时报错如下: [pcl::VoxelGrid::applyFilter] Leaf size is too small for the input d ...
【Python】解决CNN中训练权重参数不匹配size mismatch for fc.weight，size mismatch for fc.bias
目录 1.问题描述 2.问题原因 3.问题解决 3.1思路1--忽视最后一层权重额外说明:假如载入权重不写strict=False, 直接是model.load_state_dict(pre_wei ...
调试中的一些Python错误
Python出现bound method. 方法没加() PermissionError: [Errno 13] Permission denied: '...\data\\input\\dev.cs ...
YOLOV5使用过程中可能出现的问题及解决方法
我尝试过YOLOV5 v5.0 和 YOLOV5 v6.0,都出现了一些问题 (小声嘀咕:我刚开始搜索报错原因得到是github社区上作者的回答,结果把它他提供的解决方法都试了一遍,一点作用没有,最后 ...
PyTorch中的常见报错总结
Pytorch中报错报错信息非常多,这里简单总结十六种常见的报错信息,方便大家Debug 1 报错:ValueError: num_samples should be a positive integ ...

RuntimeError: Given groups=1, weight of size [1, 1, 3, 3], expected input[1, 3, 1402, 1200] to have

RuntimeError: Given groups=1, weight of size [1, 1, 3, 3], expected input[1, 3, 1402, 1200] to have相关推荐

最新文章

热门文章