更新工作目录子文件夹的序号

This commit is contained in:
2026-04-14 09:24:18 +08:00
parent 8025869b76
commit b0a94ba1e7
11 changed files with 244 additions and 255 deletions

View File

@ -534,7 +534,7 @@ if __name__ == "__main__":
print(" )")
visualizer = FlightPathVisualizer(output_dir=r"E:\code\WQ\pipeline_result\work_dir\9_visualization\flight_maps")
visualizer = FlightPathVisualizer(output_dir=r"E:\code\WQ\pipeline_result\work_dir\14_visualization\flight_maps")
# 生成飞行轨迹图
map_path = visualizer.create_flight_path_map(
gps_folder=r"D:\BaiduNetdiskDownload\20250902\gps", # GPS文件夹路径

View File

@ -2130,9 +2130,9 @@ def main():
mapper = ContentMapper()
# 示例1处理单个文件
csv_file = r"E:\code\WQ\pipeline_result\tests1\8_predictions\BGA.csv" # 采样点的预测值
csv_file = r"E:\code\WQ\pipeline_result\tests1\11_12_13_predictions\BGA.csv" # 采样点的预测值
shp_file = r"D:\BaiduNetdiskDownload\yaobao\roi\roi.shp" # 水体边界shapefile路径
output_file = r"E:\code\WQ\pipeline_result\work_dir\8_predictions\BGA.png" # 输出图片路径
output_file = r"E:\code\WQ\pipeline_result\work_dir\11_12_13_predictions\BGA.png" # 输出图片路径
#
mapper.process_data(
csv_file=csv_file,

View File

@ -79,7 +79,7 @@ class WaterQualityReportGenerator:
self.work_dir = Path(work_dir)
# 基于工作目录设置各子目录
self.visualization_dir = self.work_dir / "9_visualization"
self.visualization_dir = self.work_dir / "14_visualization"
# 设置报告保存位置默认为可视化目录visualization_dir
self._output_dir_is_default = output_dir is None
@ -585,12 +585,12 @@ class WaterQualityReportGenerator:
output_path: Optional[str] = None) -> str:
"""
生成 Word 报告 - 所有数据均来自工作目录work_dir
可视化图片、统计数据等均从 work_dir/9_visualization 和 work_dir/4_processed_data 中读取
可视化图片、统计数据等均从 work_dir/14_visualization 和 work_dir/4_processed_data 中读取
"""
# 设置工作目录(整个流程的核心)
if work_dir is not None:
self.work_dir = Path(work_dir)
self.visualization_dir = self.work_dir / "9_visualization"
self.visualization_dir = self.work_dir / "14_visualization"
if getattr(self, "_output_dir_is_default", False):
self.output_dir = self.visualization_dir
self.output_dir.mkdir(parents=True, exist_ok=True)
@ -1050,7 +1050,7 @@ class WaterQualityReportGenerator:
vis_dir = self.visualization_dir
# 0. 航线规划图
flight_path_img_path = work_dir_path / "9_visualization" / "flight_maps"
flight_path_img_path = work_dir_path / "14_visualization" / "flight_maps"
h3 = doc.add_heading("航线规划:", level=3)
self._style_heading(h3, level=3)

View File

@ -423,7 +423,7 @@ class WaterQualityVisualization:
- 2_glint文件夹单波段二值耀斑掩膜使用红色高亮显示
- 3_deglint文件夹多波段去耀斑影像使用RGB合成显示
- 自动识别文件类型并应用相应的可视化方案
- 输出保存至9_visualization/glint_deglint_previews/
- 输出保存至14_visualization/glint_deglint_previews/
Args:
work_dir: 工作目录路径
@ -713,7 +713,7 @@ class WaterQualityVisualization:
from src.postprocessing.point_map import SamplingPointMap
# 如果没有提供路径,自动查找
work_dir = self.output_dir.parent # 9_visualization的父目录就是工作目录
work_dir = self.output_dir.parent # 14_visualization的父目录就是工作目录
if hyperspectral_path is None:
# 查找高光谱影像