watermark_utils.py 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. """
  2. 水印处理兼容入口(实现已迁至 ocr_utils.watermark 包)。
  3. 新代码请优先使用::
  4. from ocr_utils.watermark import WatermarkProcessor, detect_watermark, ...
  5. 本模块保留与历史 import 路径的兼容。
  6. """
  7. from ocr_utils.watermark.algorithms import (
  8. build_watermark_mask,
  9. detect_watermark,
  10. remove_watermark_masked_adaptive,
  11. render_ratio_heatmap,
  12. save_watermark_mask_debug_layers,
  13. )
  14. from ocr_utils.watermark.contrast import (
  15. apply_contrast_enhancement_config,
  16. enhance_document_contrast,
  17. )
  18. from ocr_utils.watermark.debug import save_watermark_removal_debug
  19. from ocr_utils.watermark.pdf import (
  20. remove_txt_pdf_watermark,
  21. scan_pdf_watermark_xobjs,
  22. )
  23. from ocr_utils.watermark.removal import (
  24. remove_watermark_from_image,
  25. remove_watermark_from_image_rgb,
  26. render_watermark_mask_overlay,
  27. )
  28. __all__ = [
  29. "apply_contrast_enhancement_config",
  30. "build_watermark_mask",
  31. "detect_watermark",
  32. "enhance_document_contrast",
  33. "remove_txt_pdf_watermark",
  34. "remove_watermark_from_image",
  35. "remove_watermark_from_image_rgb",
  36. "remove_watermark_masked_adaptive",
  37. "render_ratio_heatmap",
  38. "render_watermark_mask_overlay",
  39. "save_watermark_mask_debug_layers",
  40. "save_watermark_removal_debug",
  41. "scan_pdf_watermark_xobjs",
  42. ]