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