From b2a76e23807a0b642c7129e6b3d55f3fd35f22cd Mon Sep 17 00:00:00 2001
From: tian jiax <446100073@qq.com>
Date: Wed, 15 Nov 2023 15:59:02 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E5=9C=B0=E8=A1=A8=E8=A6=86?=
 =?UTF-8?q?=E7=9B=96=E7=B1=BB=E5=9E=8B=E6=B5=81=E7=A8=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 landcover_c_sar/LandCoverMain.py | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/landcover_c_sar/LandCoverMain.py b/landcover_c_sar/LandCoverMain.py
index 6b56d066..ff55deac 100644
--- a/landcover_c_sar/LandCoverMain.py
+++ b/landcover_c_sar/LandCoverMain.py
@@ -56,8 +56,9 @@ logger = logging.getLogger("mylog")
 FILTER_SIZE = int(cf.get('filter_size'))
 file =fileHandle(DEBUG)
 MAX_TRAN_NUM = int(cf.get('max_tran__num_per_class'))
-env_str = os.path.split(os.path.realpath(__file__))[0]
-os.environ['PROJ_LIB'] = env_str
+# env_str = os.path.split(os.path.realpath(__file__))[0]
+env_str = os.path.dirname(os.path.abspath(sys.argv[0]))
+os.environ['PROJ_LIB'] = os.getcwd()
 class LandCoverMain:
     """
     算法主函数
@@ -165,6 +166,7 @@ class LandCoverMain:
         """
         预处理
         """
+
         para_names_geo = ['sim_ori']
         # for key in self.__processing_paras.keys():
         #     if "FeatureMap" in key:
@@ -544,13 +546,13 @@ class LandCoverMain:
             kernel = 'rbf'
 
         # RF
-        clf = ml.trainRF(X_train, Y_train)
-        logger.info("RF train successful")
+        # clf = ml.trainRF(X_train, Y_train)
+        # logger.info("RF train successful")
 
         # # SVM
-        # clf = ml.trainSVM(X_train, Y_train, cost, kernel)
-        # logger.info("SVM train successful")
-        # logger.info('progress bar: 60%')
+        clf = ml.trainSVM(X_train, Y_train, cost, kernel)
+        logger.info("SVM train successful")
+        logger.info('progress bar: 60%')
 
         # 生成测试集
         # X_test_path_list = ml.gene_test_set(self.__feature_tif_dir, optimal_feature)