使用python gdal从字节数据中保存图像

pip install gdal

import gdal
import numpy as np

def save_image_from_bytes(image_bytes, output_path):
    # 将字节数据转换为numpy数组
    image_array = np.frombuffer(image_bytes, dtype=np.uint8)
    
    # 获取图像的宽度和高度
    width = 512  # 替换为实际图像的宽度
    height = 512  # 替换为实际图像的高度
    
    # 创建GDAL数据集
    driver = gdal.GetDriverByName('GTiff')
    dataset = driver.Create(output_path, width, height, 3, gdal.GDT_Byte)
    
    # 将numpy数组写入GDAL数据集
    dataset.GetRasterBand(1).WriteArray(image_array.reshape((height, width)))
    dataset.GetRasterBand(2).WriteArray(image_array.reshape((height, width)))
    dataset.GetRasterBand(3).WriteArray(image_array.reshape((height, width)))
    
    # 设置地理参考信息（可选）
    dataset.SetGeoTransform((0, 1, 0, 0, 0, 1))
    dataset.SetProjection('EPSG:4326')  # 替换为实际的投影信息
    
    # 关闭数据集
    dataset = None

# 示例用法
image_bytes = b'\x00\xFF\x00\xFF...'  # 替换为实际的字节数据
output_path = 'output.tif'  # 替换为实际的输出路径
save_image_from_bytes(image_bytes, output_path)