9.将数组存储为带坐标的tif文件

最新推荐文章于 2024-07-24 11:49:51 发布

原创最新推荐文章于 2024-07-24 11:49:51 发布 · 1.5k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#python #计算机视觉 #深度学习

遥感数据处理全流程GDAL和Python 专栏收录该内容

13 篇文章

订阅专栏

本文介绍如何使用numpy数组创建带坐标的tif文件，包括读取图像属性、选择合适的数据类型和GDAL库操作，适合处理地理空间数据的转换。

1 将numpy数组存储为带坐标的tif文件

将numpy数组存储为带坐标的tif文件
def write_img(read_path, img_array):
    """
    read_path:原始文件路径
    img_array：numpy数组， 数组形状必须是(4, 36786, 37239) ,波段，行，列
    """
    read_pre_dataset = gdal.Open(read_path)
    img_transf = read_pre_dataset.GetGeoTransform()  # 仿射矩阵
    img_proj = read_pre_dataset.GetProjection()  # 地图投影信息
    print("1读取数组形状", img_array.shape,img_array.dtype.name) # (4, 36786, 37239)

    # GDT_Byte = 1, GDT_UInt16 = 2, GDT_UInt32 = 4, GDT_Int32 = 5, GDT_Float32 = 6,
    if 'uint8' in img_array.dtype.name:
        datatype = gdal.GDT_Byte
    elif 'int16' in img_array.dtype.name:
        datatype = gdal.GDT_UInt16
    else:
        datatype = gdal.GDT_Float32

    if len(img_array.shape) == 3:
        img_bands, im_height, im_width = img_array.shape
    else:
        img_bands, (im_height, im_width) = 1, img_array.shape

    filename = read_path[:-4] + '_unit8' + ".tif"
    driver = gdal.GetDriverByName("GTiff")  # 创建文件驱动
    # 注意这里先写宽再写高，对应数组的高和宽，这里不对应才对
    # https://vimsky.com/examples/detail/python-method-gdal.GetDriverByName.html
    dataset = driver.Create(filename, im_width, im_height, img_bands, datatype)
    dataset.SetGeoTransform(img_transf)  # 写入仿射变换参数
    dataset.SetProjection(img_proj)  # 写入投影

    # 写入影像数据
    if img_bands == 1:
        dataset.GetRasterBand(1).WriteArray(img_array)
    else:
        for i in range(img_bands):
            dataset.GetRasterBand(i + 1).WriteArray(img_array[i])