解决植被物候掩膜异常问题

dev
tian jiax 2024-11-13 14:45:49 +08:00
parent 46b997c132
commit 3afab435db
2 changed files with 5 additions and 3 deletions

View File

@ -10,9 +10,7 @@ import netCDF4 as Nc
import scipy.interpolate as intp import scipy.interpolate as intp
import scipy.integrate as intg import scipy.integrate as intg
from tool.algorithm.image.ImageHandle import ImageHandler from tool.algorithm.image.ImageHandle import ImageHandler
from osgeo import gdal,gdalconst
from tqdm import tqdm from tqdm import tqdm
from matplotlib import pyplot as plt
# 声明常数 # 声明常数
constdict = {} constdict = {}

View File

@ -320,12 +320,16 @@ class MachineLeaning:
:param feature_tif_dir : 特征影像路径路径 :param feature_tif_dir : 特征影像路径路径
:return X_train, Y_train : 训练数据 :return X_train, Y_train : 训练数据
""" """
cover_list = coverId.split(';')
in_tif_paths = list(glob.glob(os.path.join(feature_tif_dir, '*.tif'))) in_tif_paths = list(glob.glob(os.path.join(feature_tif_dir, '*.tif')))
land_arr = ImageHandler.get_band_array(land_cover_tif, 1) land_arr = ImageHandler.get_band_array(land_cover_tif, 1)
land_arr[np.where(np.isnan(land_arr))] = 0 land_arr[np.where(np.isnan(land_arr))] = 0
all_cover_id_list = list(np.unique(land_arr)) all_cover_id_list = list(np.unique(land_arr))
if coverId == 'empty':
cover_list = all_cover_id_list
else:
cover_list = coverId.split(';')
set1 = set(all_cover_id_list) set1 = set(all_cover_id_list)
set2 = set(cover_list) set2 = set(cover_list)