python图像分割模型怎么读取图片

Question

加载了一个模型，想放测试图片进去不想要这个全零数据，测试图片应该怎么放？能示例一下吗

Answer 1

两种方法：

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)

望采纳！谢谢