Given groups=1, weight of size [16, 1, 3, 3], expected input[1, 3, 512, 512] to have 1 channels, but got 3 channels instead
我的图片是rgb类型的,这个该怎么修改代码呀,盆友们

pytorch图像分类
- 写回答
- 好问题 0 提建议
- 关注问题
- 邀请回答
-
3条回答 默认 最新
- 瞬间的醒悟 2023-04-07 13:45关注
您可以尝试在代码中修改groups参数的值,将其从1改为3,来适配输入图片的RGB通道数,如下所示:
import torch.nn as nn class YourModel(nn.Module): def __init__(self): super(YourModel, self).__init__() self.conv = nn.Conv2d(3, 16, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=3, bias=False) def forward(self, x): x = self.conv(x) return x
这里我将groups参数从1改为3,同时修改了Conv2d的输入通道数为3(代表rgb图片的3个通道)。如果您的图片不是RGB类型的,请将输入通道数改为对应的通道数。
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报