def writeTiff(im_data, im_geotrans, im_proj, path): if 'int8' in im_data.dtype.name: datatype = gdal.GDT_Byte elif 'int16' in im_data.dtype.name: datatype = gdal.GDT_UInt16 else: datatype = gdal.GDT_Float32 if len(im_data.shape) == 3: im_bands, im_height, im_width = im_data.shape elif len(im_data.shape) == 2: im_data = np.array([im_data]) im_bands, im_height, im_width = im_data.shape # 创建文件 driver = gdal.GetDriverByName("GTiff") dataset = driver.Create(path, int(im_width), int(im_height), int(im_bands), datatype) if (dataset != None): dataset.SetGeoTransform(im_geotrans) # 写入仿射变换参数 dataset.SetProjection(im_proj) # 写入投影 for i in range(im_bands): dataset.GetRasterBand(i + 1).WriteArray(im_data[i]) del dataset解释一下
时间: 2023-05-15 13:04:38 浏览: 309
BLH.rar_BLH-XYZ_blh_blh本_site:www.pudn.com_经纬度 平面
这是一个Python函数,用于将图像数据写入TIFF文件。函数的参数包括图像数据(im_data)、地理转换信息(im_geotrans)、投影信息(im_proj)和文件路径(path)。函数首先判断图像数据的数据类型,如果是int8类型,则设置数据类型为GDT_Byte,如果是int16类型,则设置数据类型为GDT_UInt16,否则设置数据类型为GDT_Float32。接着判断图像数据的维度,如果是三维,则表示有多个波段,分别为im_bands、im_height和im_width;如果是二维,则表示只有一个波段,需要将其转换为三维数组。
阅读全文