python 栅格数据裁剪
import os
import numpy as np
import rasterio
from rasterio.windows import Window
from tqdm import tqdm
def crop_tif(image_path, image_out_path, crop_size=256):
"""
将多波段栅格影像裁剪为256*256大小的不重合小块影像
:param image_path: 待处理影像
:image_out_path: 切图保存路径
:param crop_size: 裁剪尺寸，默认为256
:return:
"""
with rasterio.open(image_path) as src:
width = src.width
height = src.height
crop_width = crop_height = crop_size
# 判断行列是否256的倍数，不是则舍弃行列的最后部分
if width % crop_size == 0:
width = width
else:
width = width - (width % crop_size)
if height % crop_size == 0:
height = height
else:
height = height - (height % crop_size)
for i in tqdm(range(0, width, crop_width), desc='影像切割256*256'):
for j in range(0, height, crop_height):
window = Window(i, j, crop_width, crop_height)
cropped_img = src.read(window=window)
# 根据窗口大小创建输出文件
out_profile = src.profile
out_profile.update({
'width': crop_width,
'height': crop_height,
'transform': src.window_transform(window)
})
# f"{'akesu'}_{i}_{j}.tif" 保存的文件名构成 可自行更改 akesu是名字，i是切割文件左上角的列序号，j是切割文件左上角的行序号
out_path = os.path.join(image_out_path, f"{'akesu'}_{i}_{j}.tif")
with rasterio.open(out_path, 'w', **out_profile) as out_ds:
out_ds.write(cropped_img)