|
@@ -65,15 +65,16 @@ class MinerUPreprocessor(BasePreprocessor):
|
|
|
|
|
|
|
|
watermark_cfg = self.config.get('watermark_removal', {})
|
|
watermark_cfg = self.config.get('watermark_removal', {})
|
|
|
wm_enabled = bool(watermark_cfg.get('enabled', False))
|
|
wm_enabled = bool(watermark_cfg.get('enabled', False))
|
|
|
|
|
+ # 对比度增强只有在水印去除之后才能生效
|
|
|
contrast_cfg = watermark_cfg.get('contrast_enhancement', {})
|
|
contrast_cfg = watermark_cfg.get('contrast_enhancement', {})
|
|
|
contrast_enabled = bool(
|
|
contrast_enabled = bool(
|
|
|
contrast_cfg.get('enabled', False) if isinstance(contrast_cfg, dict) else False
|
|
contrast_cfg.get('enabled', False) if isinstance(contrast_cfg, dict) else False
|
|
|
)
|
|
)
|
|
|
|
|
|
|
|
- if not wm_enabled and not contrast_enabled:
|
|
|
|
|
|
|
+ if not wm_enabled:
|
|
|
return image
|
|
return image
|
|
|
|
|
|
|
|
- threshold = watermark_cfg.get('threshold', 160)
|
|
|
|
|
|
|
+ threshold = watermark_cfg.get('threshold', 175)
|
|
|
morph_close_kernel = watermark_cfg.get('morph_close_kernel', 0)
|
|
morph_close_kernel = watermark_cfg.get('morph_close_kernel', 0)
|
|
|
before_image = image.copy()
|
|
before_image = image.copy()
|
|
|
try:
|
|
try:
|