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")

 

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