利用GDAL庫對tif影像進行讀取

示例代碼默認波段為[B、G、R、NIR的順序，且為四個波段]

import gdaldef readTif(fileName): dataset = gdal.Open(fileName) if dataset == None:  print(fileName+"文件無法打開")  return im_width = dataset.RasterXSize #柵格矩陣的列數 im_height = dataset.RasterYSize #柵格矩陣的行數 im_bands = dataset.RasterCount #波段數 im_data = dataset.ReadAsArray(0,0,im_width,im_height)#獲取數據 im_geotrans = dataset.GetGeoTransform()#獲取仿射矩陣信息 im_proj = dataset.GetProjection()#獲取投影信息 im_blueBand = im_data[0,0:im_height,0:im_width]#獲取藍波段 im_greenBand = im_data[1,0:im_height,0:im_width]#獲取綠波段 im_redBand = im_data[2,0:im_height,0:im_width]#獲取紅波段 im_nirBand = im_data[3,0:im_height,0:im_width]#獲取近紅外波段

寫tif影像函數

#保存tif文件函數import gdalimport numpy as npdef writeTiff(im_data,im_width,im_height,im_bands,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]) else:  im_bands, (im_height, im_width) = 1,im_data.shape  #創建文件 driver = gdal.GetDriverByName("GTiff") dataset = driver.Create(path, im_width, im_height, 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

以上這篇在python中利用GDAL對tif文件進行讀寫的方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持武林站長站。