修改地表覆盖类型流程

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)