pèlerin 2023-04-07 13:43 采纳率: 25%
浏览 9
已结题

pytorch图像分类

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类型的,这个该怎么修改代码呀,盆友们

  • 写回答

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类型的,请将输入通道数改为对应的通道数。

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(2条)

报告相同问题?

问题事件

  • 系统已结题 4月20日
  • 已采纳回答 4月12日
  • 创建了问题 4月7日

悬赏问题

  • ¥65 LineageOs-21.0系统编译问题
  • ¥30 关于#c++#的问题,请各位专家解答!
  • ¥15 App的会员连续扣费
  • ¥15 不同数据类型的特征融合应该怎么做
  • ¥15 用proteus软件设计一个基于8086微处理器的简易温度计
  • ¥15 用联想小新14Pro
  • ¥15 multisim中关于74ls192n和DSWPK开关仿真图分析(减法计数器)
  • ¥15 w3wp,exe 中发生未处理的 Microsoft ,NETFramework 异常。
  • ¥20 C51单片机程序及仿真(加减器)
  • ¥15 AQWA | 水动力分析 二阶波浪力