ocr_platform/ocr_utils/watermark/__init__.py

"""水印处理：预设、门面、算法与 PDF/调试能力。"""
from ocr_utils.watermark.algorithms import (
    build_watermark_mask,
    detect_watermark,
    remove_watermark_masked_adaptive,
    render_ratio_heatmap,
    save_watermark_mask_debug_layers,
)
from ocr_utils.watermark.contrast import (
    apply_contrast_enhancement_config,
    enhance_document_contrast,
)
from ocr_utils.watermark.debug import save_watermark_removal_debug
from ocr_utils.watermark.pdf import (
    remove_txt_pdf_watermark,
    scan_pdf_watermark_xobjs,
)
from ocr_utils.watermark.presets import (
    CELL_WATERMARK_PRESETS,
    PAGE_WATERMARK_PRESETS,
    get_preset,
    merge_watermark_config,
)
from ocr_utils.watermark.processor import WatermarkProcessor
from ocr_utils.watermark.removal import (
    remove_watermark_from_image,
    remove_watermark_from_image_rgb,
    render_watermark_mask_overlay,
)

__all__ = [
    "CELL_WATERMARK_PRESETS",
    "PAGE_WATERMARK_PRESETS",
    "WatermarkProcessor",
    "apply_contrast_enhancement_config",
    "build_watermark_mask",
    "detect_watermark",
    "enhance_document_contrast",
    "get_preset",
    "merge_watermark_config",
    "remove_txt_pdf_watermark",
    "remove_watermark_from_image",
    "remove_watermark_from_image_rgb",
    "remove_watermark_masked_adaptive",
    "render_ratio_heatmap",
    "render_watermark_mask_overlay",
    "save_watermark_mask_debug_layers",
    "save_watermark_removal_debug",
    "scan_pdf_watermark_xobjs",
]