Browse Source

fix(调整调试输出目录): 更新remove_watermark.py中的调试输出目录路径,从debug_comparison/watermark_removal/更改为debug/watermark_removal/,以统一目录结构并提升可维护性。

zhch158_admin 4 days ago
parent
commit
adafec6488
1 changed files with 4 additions and 4 deletions
  1. 4 4
      ocr_tools/remove_watermark_tool/remove_watermark.py

+ 4 - 4
ocr_tools/remove_watermark_tool/remove_watermark.py

@@ -212,7 +212,7 @@ def _maybe_save_watermark_debug(
     contrast_enhancement: Optional[Dict[str, Any]] = None,
     removal_debug: Optional[Dict[str, Any]] = None,
 ) -> None:
-    """保存调试图到 debug_comparison/watermark_removal/(与 pipeline 相同布局)。"""
+    """保存调试图到 debug/watermark_removal/(与 pipeline 相同布局)。"""
     params: Dict[str, Any] = {
         "method": settings.method,
         "threshold": settings.threshold,
@@ -298,7 +298,7 @@ def process_document(
         page_range: 页面范围字符串,如 "1-5,7,9-12"(从 1 开始,仅对 PDF 有效)
         force_image: 强制对文字型 PDF 使用图像化处理(会失去文字可搜索性,
                      但能处理水印嵌在内容流中的情况)
-        save_debug: 是否保存 before/after/compare/meta 到 debug_comparison/watermark_removal/
+        save_debug: 是否保存 before/after/compare/meta 到 debug/watermark_removal/
         debug_output_dir: 调试图根目录,默认 output_path 的父目录
         apply_watermark_removal: 默认取 settings.watermark_enabled
 
@@ -688,7 +688,7 @@ def main():
     parser.add_argument(
         "--debug",
         action="store_true",
-        help="保存调试图到 debug_comparison/watermark_removal/",
+        help="保存调试图到 debug/watermark_removal/",
     )
     parser.add_argument(
         "--debug-dir",
@@ -759,7 +759,7 @@ def main():
             logger.error(f"文件不存在: {input_path}")
             sys.exit(1)
         out_dir = args.output or (
-            input_path.parent / "debug_comparison" / "watermark_method_compare"
+            input_path.parent / "debug" / "watermark_method_compare"
         )
         paths = compare_watermark_methods(input_path, out_dir, settings)
         for k, v in paths.items():