本地磁盘A 2018-11-08 11:17 采纳率: 100%
浏览 4317
已结题

萌新求问,我神经网络书上的一段代码出错不知道为什么??

这是陈云的《深度学习框架Pytorch入门与实践》4.2.1节图像相关层的代码:

from PIL import Image
from torchvision.transforms import ToTensor,ToPILImage
import torch as t
from torch import nn
from torch.autograd import Variable as V
to_tensor=ToTensor()
to_pil=ToPILImage()
lena=Image.open('C:/Users/Desktop/lena.png')
input=to_tensor(lena).unsqueeze(0)

kernel=t.ones(3,3)/-9
kernel[1][1]=1
conv=nn.Conv2d(1,1,(3,3),1,bias=False)
conv.weight.data=kernel.view(1,1,3,3)

out=conv(V(input))
to_pil(out.data.squeeze(0))

但是结果报错:

File "D:\py\lib\site-packages\torch\nn\modules\conv.py", line 301, in forward
self.padding, self.dilation, self.groups)

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

这怎么解决啊。。。。。

  • 写回答

2条回答

  • xwang71785 2019-03-10 19:58
    关注

    lena=Image.open('C:/Users/Desktop/lena.png')
    我猜你用的lena.png是彩色的(所以是3个channels)
    下载一个黑白的试试(只有1个channel)

    评论

报告相同问题?

悬赏问题

  • ¥15 求差集那个函数有问题,有无佬可以解决
  • ¥15 【提问】基于Invest的水源涵养
  • ¥20 微信网友居然可以通过vx号找到我绑的手机号
  • ¥15 寻一个支付宝扫码远程授权登录的软件助手app
  • ¥15 解riccati方程组
  • ¥15 display:none;样式在嵌套结构中的已设置了display样式的元素上不起作用?
  • ¥15 使用rabbitMQ 消息队列作为url源进行多线程爬取时,总有几个url没有处理的问题。
  • ¥15 Ubuntu在安装序列比对软件STAR时出现报错如何解决
  • ¥50 树莓派安卓APK系统签名
  • ¥65 汇编语言除法溢出问题