【Python&RS】遥感影像的像素坐标转地理坐标（仿射变换）

def Get_data(filepath):
    """
    :param filepath: 输入影像的路径
    :return: 返回仿射地理变换参数
    """
    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))
    return ds_geo
    # data = ds.ReadAsArray(0, 0, ds_width, ds_height)  # 以数组的形式读取整个数据集

def Pixel_Coordinate(x, y, ds_geo):
    """
    :param x: 输入x像素坐标
    :param y: 输入y像素坐标
    :param ds_geo: 输入仿射地理变换参数
    :return: 返回x,y的地理/投影坐标
    """
    x_geo = ds_geo[0]+ds_geo[1]*x + y*ds_geo[2]
    y_geo = ds_geo[3]+ds_geo[4]*x + y*ds_geo[5]
    print(x_geo, y_geo)
    return x_geo, y_geo

def Coordinate_Pixel(x_geo, y_geo, ds_geo):
    """
    :param x_geo: 输入x地理坐标
    :param y_geo: 输入y地理坐标
    :param ds_geo: 输入仿射地理变换参数
    :return: 返回x,y像素坐标
    """
    y = ((y_geo - ds_geo[3] - ds_geo[4] / ds_geo[1] * x_geo + ds_geo[4] / ds_geo[1] * ds_geo[
        0]) / (ds_geo[5] - ds_geo[4] / ds_geo[1] * ds_geo[2]))
    x = ((x_geo - ds_geo[0] - y * ds_geo[2]) / ds_geo[1])
    return int(x), int(y)