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

在使用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")

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章