基于Python批量镶嵌拼接遥感影像/栅格数据(示例代码)

from osgeo import gdal

def Get_data(filepath):
    ds = gdal.Open(filepath)  # 打开数据集dataset
    ds_width = ds.RasterXSize  # 获取数据宽度
    ds_height = ds.RasterYSize  # 获取数据高度
    ds_bands = ds.RasterCount  # 获取波段数
    ds_geo = ds.GetGeoTransform()  # 获取仿射地理变换参数
    ds_prj = ds.GetProjection()  # 获取投影信息
    print("影像的宽度为：" + str(ds_width))
    print("影像的高度为：" + str(ds_height))
    print("仿射地理变换参数为：" + str(ds_geo))
    print("投影坐标系为：" + str(ds_prj))
    # data = ds.ReadAsArray(0, 0, ds_width, ds_height)  # 以数组的形式读取整个数据集

def Mosaic_GDAL(path_image):
    """
    :param path_image: 需要镶嵌影像的路径
    :return: None
    """
    path = path_image
    path_lists = os.listdir(path)
    for i in range(0, len(path_lists)):
        print("正在处理第%s幅影像......" % i)
        if i == 0:
            continue
        elif i == 1:
            img1 = gdal.Open(path+path_lists[0], gdal.GA_ReadOnly)
            img2 = gdal.Open(path+path_lists[1], gdal.GA_ReadOnly)
            input_proj = img2.GetProjection()
            options = gdal.WarpOptions(srcSRS=input_proj, dstSRS=input_proj, format='GTiff',
                                       resampleAlg=gdal.GRA_NearestNeighbour, callback=Show_Progress)
            # 输入投影，输出投影，输出格式，重采样方法
            gdal.Warp(path+"%s.tif" % (i+1), [img1, img2], options=options)  # 输出路径，需要镶嵌的数据，参数配置
            img1 = None
            img2 = None
            del img1, img2
        else:
            img1 = gdal.Open(path+"%s.tif" % i, gdal.GA_ReadOnly)
            img2 = gdal.Open(path + path_lists[i], gdal.GA_ReadOnly)
            input_proj = img2.GetProjection()
            options = gdal.WarpOptions(srcSRS=input_proj, dstSRS=input_proj, format='GTiff',
                                       resampleAlg=gdal.GRA_NearestNeighbour, callback=Show_Progress)
            # 输入投影，输出投影，输出格式，重采样方法
            gdal.Warp(path + "%s.tif" % (i + 1), [img1, img2], options=options)  # 输出路径，需要镶嵌的数据，参数配置
            img1 = None
            img2 = None
            del img1, img2