bug解决：Given groups=1, weight of size 6 1 5 5, expected input[4, 3, 224, 224] to have 1 channels, but

原创

于 2020-06-08 22:49:58 发布 · 7k 阅读

CC 4.0 BY-SA版权

文章标签：

博客讨论了一个在处理计算机视觉任务时遇到的bug，涉及将图像从三通道转换为一通道的灰度图。通过修改代码，使用transforms.Grayscale函数，作者发现预期的输入通道数与实际提供的不匹配。问题在于将三通道图像转换为灰度图的正确方法并未得到应用，导致代码无法正常工作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given groups=1, weight of size 6 1 5 5, expected input[4, 3, 224, 224] to have 1 channels, but got 3 channels instead

代码原本为：

200万优质内容无限畅学