python GDAL和PIL图像转换

读取数据

先来对比一下GDAL和PIL数据读取的结果
GDAL：

ds: gdal.Dataset = gdal.Open('test.jpg')
gdal_arr = ds.ReadAsArray()
print(gdal_arr.shape)

# (3, 256, 256)

PIL：

img = Image.open('test.jpg')
pil_arr = np.array(img)
print(pil_arr.shape)

# (256, 256, 3)

转换

发现GDAL和PIL的图像存储方式不太一样，相互转换方法如下：

# gdal_arr 转 pil_arr
array= np.rollaxis(gdal_array , 0, 3)

# pil_arr 转 gdal_arr
array = np.rollaxis(pil_array , 2, 0)

最后就可以使用GDAL和PIL相关方法读取数据，如下：

from osgeo import gdal_array

# PIL 要求array.shape为(256, 256, 3)
img = Image.fromarray(array, mode='RGB')

# GDAL 要求array.shape为(3, 256, 256)
ds = gdal_array.OpenArray(array)