import gdal
import cv2
import numpy as np#读取tif数据集
def readTif(fileName):dataset = gdal.Open(fileName)if dataset == None:print(fileName+"文件无法打开")return dataset#获取投影信息
def Getproj(fileName):dataset = readTif(fileName)return dataset.GetProjection()#获取仿射矩阵信息
def Getgeotrans(fileName):dataset = readTif(fileName)return dataset.GetGeoTransform()#opencv数据转gdal
def OpencvData2GdalData(OpencvImg_data):# 若为二维，格式相同if(len(OpencvImg_data.shape) == 2):GdalImg_data = OpencvImg_dataelse:if 'int8' in OpencvImg_data.dtype.name:GdalImg_data = np.zeros((OpencvImg_data.shape[2],OpencvImg_data.shape[0],OpencvImg_data.shape[1]),np.uint8)elif 'int16' in OpencvImg_data.dtype.name:GdalImg_data = np.zeros((OpencvImg_data.shape[2],OpencvImg_data.shape[0],OpencvImg_data.shape[1]),np.uint16)else:GdalImg_data = np.zeros((OpencvImg_data.shape[2],OpencvImg_data.shape[0],OpencvImg_data.shape[1]),np.float32)for i in range(OpencvImg_data.shape[2]):# 注意，opencv为BGRdata = OpencvImg_data[:,:,OpencvImg_data.shape[2]-i-1]data = np.reshape(data,(OpencvImg_data.shape[0],OpencvImg_data.shape[1]))GdalImg_data[i] = datareturn GdalImg_data#opencv数据转gdal
def GdalData2OpencvData(GdalImg_data):if 'int8' in GdalImg_data.dtype.name:OpencvImg_data = np.zeros((GdalImg_data.shape[1],GdalImg_data.shape[2],GdalImg_data.shape[0]),np.uint8)elif 'int16' in GdalImg_data.dtype.name:OpencvImg_data = np.zeros((GdalImg_data.shape[1],GdalImg_data.shape[2],GdalImg_data.shape[0]),np.uint16)else:OpencvImg_data = np.zeros((GdalImg_data.shape[1],GdalImg_data.shape[2],GdalImg_data.shape[0]),np.float32)for i in range(GdalImg_data.shape[0]):OpencvImg_data[:,:,i] = GdalImg_data[GdalImg_data.shape[0]-i-1,:,:]return OpencvImg_data#保存tif文件函数
def writeTiff(im_data,im_geotrans,im_proj,path):if 'int8' in im_data.dtype.name:datatype = gdal.GDT_Byteelif 'int16' in im_data.dtype.name:datatype = gdal.GDT_UInt16else:datatype = gdal.GDT_Float32if len(im_data.shape) == 3:im_bands, im_height, im_width = im_data.shapeelif 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# 测试OpenCV转gdal
proj = Getproj("1.tif")
geotrans = Getgeotrans("1.tif")
OpencvImg_data = cv2.imread("1.tif")
GdalImg_data = OpencvData2GdalData(OpencvImg_data)
writeTiff(GdalImg_data,geotrans,proj,"2.tif")
# 测试gdal转OpenCV
dataset = readTif("1.tif")
width = dataset.RasterXSize #栅格矩阵的列数
height = dataset.RasterYSize #栅格矩阵的行数
GdalImg_data = dataset.ReadAsArray(0,0,width,height)#获取数据
OpencvImg_data = GdalData2OpencvData(GdalImg_data)
cv2.imwrite("3.tif",OpencvImg_data)