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

文章目录

• 一、image与numpy相互转换
• • image转numpy
  • numpy转image
• 二、numpy与tensor相互转换
• • numpy转tensor
  • tensor转numpy
• 三、图像加载 & 转换 & 保存
• • 图像加载
  • 单通道图像转换多通道
  • 图像保存
• 四、numpy切片 注意事项

一、image与numpy相互转换 image转numpy

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

numpy转image

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

二、numpy与tensor相互转换 numpy转tensor

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

tensor转numpy

变量名 = 变量名.numpy()

三、图像加载 & 转换 & 保存 图像加载

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

单通道图像转换多通道

变量名 = 变量名.convert(‘RGB’)
此处’RGB’只是其中一种格式

图像保存

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

四、numpy切片 注意事项

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

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

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

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