diff --git a/atmosphericDelay-C-SAR/AtmosphericDelayTool.py b/atmosphericDelay-C-SAR/AtmosphericDelayTool.py index 20285559..5b6b0379 100644 --- a/atmosphericDelay-C-SAR/AtmosphericDelayTool.py +++ b/atmosphericDelay-C-SAR/AtmosphericDelayTool.py @@ -10,9 +10,7 @@ import netCDF4 as Nc import scipy.interpolate as intp import scipy.integrate as intg from tool.algorithm.image.ImageHandle import ImageHandler -from osgeo import gdal,gdalconst from tqdm import tqdm -from matplotlib import pyplot as plt # 声明常数 constdict = {} diff --git a/tool/algorithm/ml/machineLearning.py b/tool/algorithm/ml/machineLearning.py index 55e784a6..0cb4898a 100644 --- a/tool/algorithm/ml/machineLearning.py +++ b/tool/algorithm/ml/machineLearning.py @@ -320,12 +320,16 @@ class MachineLeaning: :param feature_tif_dir : 特征影像路径路径 :return X_train, Y_train : 训练数据 """ - cover_list = coverId.split(';') 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[np.where(np.isnan(land_arr))] = 0 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) set2 = set(cover_list)