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

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 = {}

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)