####计算两幅tif图像的MAE import torch from osgeo import gdal import numpy as np import matplotlib.pyplot as plt import rasterio data1=gdal.Open(r'D:\遥感数据集\CIA\val\2002_076_0316-2002_092_0401\L71093084_08420020401_HRF_modtran_surf_ref_agd66.tif') data2=gdal.Open(r'D:\save\test\PRED_2002_076_0316-2002_092_0401.tif') tmp_img1=data1.ReadAsArray() tmp_img2=data2.ReadAsArray() print(tmp_img1.shape) #(6,2040,1720) print(tmp_img2.shape) #(6,1792,1280) #将原始图像进行裁剪 real_img1=tmp_img1[:,:1792,:1280] print(real_img1.shape) print(type(real_img1)) img1_tensor=torch.tensor(real_img1) img2_tensor=torch.tensor(tmp_img2) #计算绝对误差 absolute_error=torch.abs(img1_tensor-img2_tensor) print(absolute_error) # 计算 MAE(平均绝对误差) mae = torch.mean(absolute_error.float()) print("MAE:",mae.item())
遥感图像实验,衡量时空融合算法的MAE
最新推荐文章于 2024-10-27 11:29:42 发布