pytorch summary RuntimeError, 预期通道不匹配。

问题描述 投票:-1回答:1

我是AI的新手,当我试图从我的网络中获取摘要信息时,我得到了RuntimeError。this 根据错误信息,我应该传递什么参数? 谢谢。

image-processing computer-vision pytorch artificial-intelligence
1个回答
0
投票

你的第一层期望有3个通道的输入(在图像中,这通常意味着有红、绿、蓝通道的彩色图像),但你的输入似乎只有一个通道(在图像中,这通常意味着你的输入图像是灰度图像而不是彩色图像)。

© www.soinside.com 2019 - 2024. All rights reserved.