6.0 深度学习图片常见转换操作

一、image与numpy相互转换

image转numpy

变量名 = numpy.array(变量名)

numpy转image

变量名 = Image.fromarray(变量名)

二、numpy与tensor相互转换

numpy转tensor

变量名 = torch.tensor(变量名)

tensor转numpy

变量名 = 变量名.numpy()

三、图像加载与保存

图像加载

变量名 = Image.open(‘文件路径/文件名’)

图像保存

变量名.save(‘文件路径/文件名’)

四、numpy切片 注意事项

1、变量名[…, …, …]
对矩阵进行一次切片

2、变量名 […] […]…[…]
对矩阵进行迭代切片

例：
arr1 的形状为4 * 5 * 6
arr1[: ,: , 0] 的形状为 4 * 5
arr1[:] [:] [0] = arr1[0] 的形状为 5 * 6

解释：[]符号迭代切片后者前两个[]都是取全体元素只有最后一个[0]有效