赞
踩
@影像nan转换为0
做高光谱降维处理或是纹理信息特征提取都会经历这一步,当小编看到进度条到达99%时,内心非常喜悦,紧接着却弹出下面的对话框,一盆冷水直淋头皮。于是乎,去网上查方法,查来查去,方法就那几种,而且都不能很好亦或是很方便的解决。
nan意为:not a number ,影像中出现nan值得情况有好几种,譬如:数字图像处理结果导出到本地时、数据格式问题、还有就是原始数据就他喵的这样,小编飞的无人机数据就是第三种情况。
罢了,我累了,我也妥协了。于是乎,我凭借着可怜的基础自己写了一组python代码来将整张影像中nan值替换为0,并将原始影像输出,保留原始影像的任何元数据信息。其PCA结果如下:
需要的库:gdal、numpy、pandas
from osgeo import gdal import numpy as np import os import pandas as pd #读取tif数据集 def readTif(path): dataset = gdal.Open(path) if dataset == None: print(fileName+"文件无法打开") return dataset UAV_Path = r"F:\uavdlw\20210516lxjian(5)\lx.tif" UAV_Path = UAV_Path.encode('gbk','ignore') SavePath = r"F:\uavdlw\20210516lxjian(5)\lxp.tif" # 读取影像 uav_data = readTif(UAV_Path) Tif_width = uav_data.RasterXSize #栅格矩阵的列数 Tif_height = uav_data.RasterYSize #栅格矩阵的行数 Tif_geotrans = uav_data.GetGeoTransform()#获取仿射矩阵信息 Tif_proj = uav_data.GetProjection()#获取投影信息 uav_data = uav_data.ReadAsArray(0,0,Tif_width,Tif_height) #保存tif文件函数 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 uav_data=np.nan_to_num(uav_data)#nan替换为0 uav_data.shape#通道5,行数列数 writeTiff(uav_data,Tif_geotrans,Tif_proj,SavePath)
呼~~~~,完结。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。