解决植被物候掩膜异常问题
parent
46b997c132
commit
3afab435db
|
@ -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 = {}
|
||||||
|
|
|
@ -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)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue