用gdal分块读取和裁剪卫星影像

起因

最近需要将高分-2的高分辨率影像裁剪成小块用于制作深度学习的训练样本，因而研究了一下如何使用python 代码来批量对影像进行切块。

这里主要需要使用到的是gdal和numpy两个包

通过如下代码可以安装这两个包：

pip install gdal
pip install numpy
#or
conda install gdal
conda install numpy

导入相应的库

from osgeo import gdal
import numpy as np
import os
import random
from tqdm import tqdm

获取待处理的影像的路径

#step1: read the image and set the outpath

# 分块影像所在文件夹，不能有中文
tifDir = r"E:\GF-image"
# 输出的文件夹，不能有中文，如果文件夹不存在则会被创建
outPath = r"E:\GF-imageout"
if not os.path.exists(outPath):
    os.makedirs(outPath)
#将tif影像都找出来得到一个列表 
tifs = [i for i in os.listdir(tifDir) if i.endswith(".tif")]
 
print("有 %s 个tif文件" % len(tifs))
print("tifs",tifs)

边读取边导出

我这里设定导出影像大小为256*256,故size = 256.

# step2: clip the image to 256x256
# 定义切图的大小（矩形框）
size = 256
for img in rang