将图片读进网络可能会因为图片通道不为而报错,这里总结了不同通道处理的情况
单通道转3通道
from PIL import Image
import numpy as np
im = np.asarray(Image.open(path))
if len(im) == 2:
c = []
for i in range(3):
c.append(im)
im = np.asarray(c)
im = im.transpose([1,2,0])
4通道转3通道
from PIL import Image
if im.shape[2] == 4:
im = Image.open(im_path).convert("RGB")