|
@@ -1402,8 +1402,8 @@ def save_watermark_removal_debug(
|
|
|
"""
|
|
"""
|
|
|
保存去水印调试图(before / after / compare / meta.json)。
|
|
保存去水印调试图(before / after / compare / meta.json)。
|
|
|
|
|
|
|
|
- 与 universal_doc_parser 的 debug_comparison 目录结构一致:
|
|
|
|
|
- ``{output_dir}/debug_comparison/{subdir}/``
|
|
|
|
|
|
|
+ 与 universal_doc_parser 的 module debug 目录结构一致:
|
|
|
|
|
+ ``{output_dir}/debug/{subdir}/``
|
|
|
|
|
|
|
|
Args:
|
|
Args:
|
|
|
before: 处理前图像(RGB/BGR/灰度)
|
|
before: 处理前图像(RGB/BGR/灰度)
|
|
@@ -1413,7 +1413,7 @@ def save_watermark_removal_debug(
|
|
|
processing_params: 写入 meta.json 的参数(threshold、contrast_enhancement 等)
|
|
processing_params: 写入 meta.json 的参数(threshold、contrast_enhancement 等)
|
|
|
image_format: 图片格式,png/jpg
|
|
image_format: 图片格式,png/jpg
|
|
|
save_compare: 是否保存左右拼接对比图
|
|
save_compare: 是否保存左右拼接对比图
|
|
|
- subdir: debug_comparison 下的子目录名
|
|
|
|
|
|
|
+ subdir: debug 根目录下的子目录名(默认 watermark_removal)
|
|
|
|
|
|
|
|
Returns:
|
|
Returns:
|
|
|
已保存文件路径字典(before/after/compare/meta,未保存的键省略)
|
|
已保存文件路径字典(before/after/compare/meta,未保存的键省略)
|
|
@@ -1423,9 +1423,9 @@ def save_watermark_removal_debug(
|
|
|
if isinstance(after, Image.Image):
|
|
if isinstance(after, Image.Image):
|
|
|
after = np.array(after)
|
|
after = np.array(after)
|
|
|
|
|
|
|
|
- root = Path(output_dir)
|
|
|
|
|
- debug_dir = root / "debug_comparison" / subdir
|
|
|
|
|
- debug_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
|
|
|
+ from ocr_utils.module_debug_viz import resolve_module_debug_dir
|
|
|
|
|
+
|
|
|
|
|
+ debug_dir = resolve_module_debug_dir(output_dir, subdir)
|
|
|
|
|
|
|
|
fmt = (image_format or "png").lstrip(".")
|
|
fmt = (image_format or "png").lstrip(".")
|
|
|
before_bgr = _image_to_bgr_for_debug(before)
|
|
before_bgr = _image_to_bgr_for_debug(before)
|