RuntimeError: Given groups=1, weight of size 32 1 3 3, expected input[5, 3, 224, 224] to have 1 chan

RuntimeError: Given groups=1, weight of size 32 1 3 3, expected input[5, 3, 224, 224] to have 1 channels, but got 3 channels instead解决办法

这个报错的意思是,预期输入的图像通道是1,但是得到了3个通道的图像,所以在图片输入之前要把图像的通道变化一下

from PIL import Image
image=Image.open(r"图片路径")
image=image.convert("L")

你可能感兴趣的:(笔记,pytorch,tensorflow,机器学习,神经网络,深度学习)