通过Python中的块加载图像以进行GDAL处理

Question

我需要加载图像*.tif以便在GDAL中处理它。我使用这段代码：

data = gdal.Open("a.tif", gdal.GA_ReadOnly)
img = data.ReadAsArray()

一切都适用于小图像（约10Mb）。但是当我尝试加载更大的图像（~1Gb）时，它开始使用大量内存（~15Gb RAM）。 如何通过块加载此图像进行顺序处理？

谢谢！

Answer 1

ReadAsArray()有一些可选参数来读取图像的各个部分。

ReadAsArray(x_off, y_off, x_size, y_size)

完整代码：

import gdal
ds = gdal.Open('input.tif', gdal.GA_ReadOnly)
rb = ds.GetRasterBand(1)
xsize = rb.XSize
ysize = rb.YSize
ystep = ysize / 10
yresidual = ysize - (ystep * 10)

for i in range(10):
    if i != 9:
        img_part = rb.ReadAsArray(0, ystep * i, xsize, ystep)
    else:
        img_part = rb.ReadAsArray(0, ystep * i, xsize, ystep + yresidual)
    # do something with img_part

ds = None