Mega-1.1 全链路路径归一化收尾(18 文件)

This commit is contained in:
DXC
2026-06-15 15:20:50 +08:00
parent a9e77d2ad0
commit 82e0b92af6
18 changed files with 69 additions and 69 deletions

View File

@ -11,7 +11,7 @@
5. 模型训练摘要报告training_summary.csv
6. 参数反演结果报告:包含预测统计信息
7. 批量处理摘要batch_inference_summary.json
8. 掩膜和耀斑缩略图2_glint和3_deglint文件夹的影像预览图
8. 掩膜和耀斑缩略图2_Glint_Detection和3_deglint文件夹的影像预览图
"""
import numpy as np
@ -423,10 +423,10 @@ class WaterQualityVisualization:
generate_glint: bool = True,
generate_deglint: bool = True) -> Dict[str, str]:
"""
根据工作目录的2_glint和3_deglint文件夹中的文件生成PNG预览图
根据工作目录的2_Glint_Detection和3_deglint文件夹中的文件生成PNG预览图
功能特点:
- 2_glint文件夹单波段二值耀斑掩膜使用红色高亮显示
- 2_Glint_Detection文件夹:单波段二值耀斑掩膜,使用红色高亮显示
- 3_deglint文件夹多波段去耀斑影像使用RGB合成显示
- 自动识别文件类型并应用相应的可视化方案
- 输出保存至14_visualization/glint_deglint_previews/
@ -434,7 +434,7 @@ class WaterQualityVisualization:
Args:
work_dir: 工作目录路径
output_subdir: 输出子目录名称(默认 "glint_deglint_previews"
generate_glint: 是否处理2_glint文件夹中的文件
generate_glint: 是否处理2_Glint_Detection文件夹中的文件
generate_deglint: 是否处理3_deglint文件夹中的文件
Returns:
@ -461,17 +461,17 @@ class WaterQualityVisualization:
print(f"{'='*60}")
print(f"输出目录: {output_dir}")
# 处理2_glint文件夹
# 处理2_Glint_Detection文件夹
if generate_glint:
glint_dir = work_dir_path / "2_glint"
glint_dir = work_dir_path / "2_Glint_Detection"
if glint_dir.exists():
print(f"正在处理2_glint文件夹: {glint_dir}")
print(f"正在处理2_Glint_Detection文件夹: {glint_dir}")
glint_previews = self._process_image_folder(
glint_dir, output_dir, "glint", preview_paths
)
processed_count += len(glint_previews)
else:
print(f"警告: 2_glint文件夹不存在: {glint_dir}")
print(f"警告: 2_Glint_Detection文件夹不存在: {glint_dir}")
# 处理3_deglint文件夹
if generate_deglint:
@ -536,7 +536,7 @@ class WaterQualityVisualization:
为可视化模块生成影像预览图
特别处理:
- 耀斑掩膜 (2_glint/*.dat)单波段二值图黑底0、耀斑区域为白1
- 耀斑掩膜 (2_Glint_Detection/*.dat)单波段二值图黑底0、耀斑区域为白1
- 其他影像多波段RGB合成使用波长选择RGB波段
Args: