python pytorch模型中出现输入错误

imzjd6km 于 2023-02-18 发布在 Python

关注(0)|答案(1)|浏览(270)

当我尝试在pytorch中执行一个模型时，我得到了以下错误。
Given groups=1, weight of size [64, 3, 4, 4], expected input[1, 4, 512, 512] to have 3 channels, but got 4 channels instead
我知道我给模型的输入图像有4个通道，而它需要有3个通道。但是，有人能告诉我括号中每个数字的含义是什么吗？我可以从哪里开始调试这个问题？因为我给模型输入了2种类型的图像和图像标签。
如果您需要更多的信息，请发表意见。我将很乐意提供。

python

来源：https://stackoverflow.com/questions/75482612/getting-wrong-input-error-in-pytorch-model

1条答案

按热度按时间

cqoc49vn1#

内核组的大小为[64,3,4,4] -

该层中有64个过滤器或内核
每个内核3个通道（应与每个图像的通道匹配）
内核大小（高度和宽度）分别为4和4

在您的图像中，如注解中所述-[1,4,512,512] = [批量大小，通道，高度，宽度]。
要解决这个问题，您可以从映像中删除一个通道，或者向内核的1维添加一个额外的通道，如下所示：

model.conv1.shape #  [64,3,4,4]

new_conv_layer = torch.nn.Conv2d(4, 64, kernel_size=4)

# initialize first 3 channels with values from old layer
new_conv_layer.weight.data[:,:3,:,:] = model.conv1.weight.data.clone()

# initialize new channel however you want
new_conv_layer.weight.data[:,3,:,:] =  ... 

# assign replacement layer 
model.conv1 = new_conv_layer

model.conv1.shape  # [64,4,4,4]

赞(0）回复(0）举报 2023-02-18

我来回答

python pytorch模型中出现输入错误

1条答案

相关问题

热门标签

最新问答