更新工作目录子文件夹的序号
This commit is contained in:
@ -246,8 +246,8 @@ def non_empirical_retrieval(algorithm, model_info_path, coor_spectral_path, outp
|
||||
|
||||
if __name__ == "__main__":
|
||||
algorithm= "chl_a"
|
||||
model_info_path= r"E:\code\WQ\pipeline_result\work_dir\5_training_spectra\6_5_non_empirical_models\SS\SS_chl_a.json"
|
||||
coor_spectral_path= r"E:\code\WQ\pipeline_result\work_dir\7_sampling\sampling_spectra.csv"
|
||||
output_path= r"E:\code\WQ\pipeline_result\work_dir\8_predictions\SS_chl_a.csv"
|
||||
model_info_path= r"E:\code\WQ\pipeline_result\work_dir\5_training_spectra\8_non_empirical_models\SS\SS_chl_a.json"
|
||||
coor_spectral_path= r"E:\code\WQ\pipeline_result\work_dir\10_sampling\sampling_spectra.csv"
|
||||
output_path= r"E:\code\WQ\pipeline_result\work_dir\11_12_13_predictions\SS_chl_a.csv"
|
||||
wave_radius=5.0
|
||||
non_empirical_retrieval(algorithm, model_info_path, coor_spectral_path, output_path, wave_radius)
|
||||
@ -95,11 +95,11 @@ class WaterQualityInversionPipeline:
|
||||
self.deglint_dir = self.work_dir / "3_deglint"
|
||||
self.processed_data_dir = self.work_dir / "4_processed_data"
|
||||
self.training_spectra_dir = self.work_dir / "5_training_spectra"
|
||||
self.models_dir = self.work_dir / "6_models"
|
||||
self.sampling_dir = self.work_dir / "7_sampling"
|
||||
self.prediction_dir = self.work_dir / "8_predictions"
|
||||
self.visualization_dir = self.work_dir / "9_visualization"
|
||||
self.reports_dir = self.work_dir / "10_reports"
|
||||
self.models_dir = self.work_dir / "7_models"
|
||||
self.sampling_dir = self.work_dir / "10_sampling"
|
||||
self.prediction_dir = self.work_dir / "11_12_13_predictions"
|
||||
self.visualization_dir = self.work_dir / "14_visualization"
|
||||
self.reports_dir = self.work_dir / "reports"
|
||||
|
||||
# 创建所有子目录
|
||||
for dir_path in [self.water_mask_dir, self.glint_dir, self.deglint_dir,
|
||||
|
||||
@ -105,14 +105,14 @@ class WaterQualityInversionPipeline:
|
||||
self.deglint_dir = self.work_dir / "3_deglint"
|
||||
self.processed_data_dir = self.work_dir / "4_processed_data"
|
||||
self.training_spectra_dir = self.work_dir / "5_training_spectra"
|
||||
self.indices_dir = self.work_dir / "5_5_water_quality_indices"
|
||||
self.models_dir = self.work_dir / "6_models"
|
||||
self.non_empirical_models_dir = self.work_dir / "6_5_non_empirical_models"
|
||||
self.custom_regression_dir = self.work_dir / "6_75_custom_regression"
|
||||
self.sampling_dir = self.work_dir / "7_sampling"
|
||||
self.prediction_dir = self.work_dir / "8_predictions"
|
||||
self.visualization_dir = self.work_dir / "9_visualization"
|
||||
self.reports_dir = self.work_dir / "10_reports"
|
||||
self.indices_dir = self.work_dir / "6_water_quality_indices"
|
||||
self.models_dir = self.work_dir / "7_models"
|
||||
self.non_empirical_models_dir = self.work_dir / "8_non_empirical_models"
|
||||
self.custom_regression_dir = self.work_dir / "9_custom_regression"
|
||||
self.sampling_dir = self.work_dir / "10_sampling"
|
||||
self.prediction_dir = self.work_dir / "11_12_13_predictions"
|
||||
self.visualization_dir = self.work_dir / "14_visualization"
|
||||
self.reports_dir = self.work_dir / "reports"
|
||||
|
||||
# 创建所有子目录
|
||||
for dir_path in [self.water_mask_dir, self.glint_dir, self.deglint_dir,
|
||||
@ -3379,10 +3379,10 @@ class WaterQualityInversionPipeline:
|
||||
else:
|
||||
# 如果output_dir为空,使用工作目录
|
||||
if hasattr(self, 'work_dir') and self.work_dir is not None:
|
||||
non_empirical_dir = Path(self.work_dir) / "6_5_non_empirical_models"
|
||||
non_empirical_dir = Path(self.work_dir) / "8_non_empirical_models"
|
||||
else:
|
||||
# 如果没有工作目录,使用当前目录
|
||||
non_empirical_dir = Path.cwd() / "6_5_non_empirical_models"
|
||||
non_empirical_dir = Path.cwd() / "8_non_empirical_models"
|
||||
non_empirical_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# 设置默认参数
|
||||
@ -3712,7 +3712,7 @@ class WaterQualityInversionPipeline:
|
||||
if non_empirical_models_dir is not None:
|
||||
final_models_dir = non_empirical_models_dir
|
||||
else:
|
||||
default_models_dir = str(self.work_dir / "6_5_non_empirical_models")
|
||||
default_models_dir = str(self.work_dir / "8_non_empirical_models")
|
||||
if Path(default_models_dir).exists():
|
||||
final_models_dir = default_models_dir
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user