@影像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)
呼~~~~,完结。