两种方法:
from PIL import Image
image_1 = Image.open(img_path)
from skimage import io
image_2 = io.imread(img_path)
#升维
source_image = np.expand_dims(source_image.transpose((2,0,1)),0)
#转为tensor张量
source_image = torch.Tensor(source_image)
望采纳!谢谢
您好,我是有问必答小助手,您的问题已经有小伙伴帮您解答,感谢您对有问必答的支持与关注!