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的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。