RuntimeError: Given groups=1, weight of size 64 4 7 7, expected input[1, 5, 206, 206]错误

2024年6月9日 717次阅读

在使用python进行图像机器学习的时候，由于输入图片的问题会报错：

RuntimeError: Given groups=1, weight of size 64 4 7 7, expected input[1, 5, 206, 206] to have 4 channels, but got 5 channels instead
 

类似的错误，问题主要是出在，输入图片不是标准的RGB图片，这时候需要将图片转换为RGB图片，代码如下：

from PIL import Image

L_path='./test_img/original/timg.png'
L_image=Image.open(L_path)
out = L_image.convert("RGB")
out.save("./test_img/sg/timg_rgb.png")

 

点赞