From 3afab435db20f18fc6d838483c4c1a4e41e93afe Mon Sep 17 00:00:00 2001
From: tian jiax <446100073@qq.com>
Date: Wed, 13 Nov 2024 14:45:49 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3=E6=A4=8D=E8=A2=AB=E7=89=A9?=
 =?UTF-8?q?=E5=80=99=E6=8E=A9=E8=86=9C=E5=BC=82=E5=B8=B8=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 atmosphericDelay-C-SAR/AtmosphericDelayTool.py | 2 --
 tool/algorithm/ml/machineLearning.py           | 6 +++++-
 2 files changed, 5 insertions(+), 3 deletions(-)

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)