深度学习修改通道数从1到2

这个bug改不了,是输入的input有问题吧?那这个怎么改呀

img

大概就是实际的输入是[64,1,28,28],就是64张分辨率为28×28的单通道图像,但是你的卷积核是[64,2,3,3],也就是卷积核是有两个通道的,网络期望你输入两通道的图像,而不是单通道。可以用通道拼接等方式改变通道数,具体就取决于你做的是什么问题,用在什么应用环境上面了。

不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^