configuration_detector.py

"""Config for the standalone Cage watermark detector."""
from .cage_config import CageExtractorConfig


class CageDetectorConfig(CageExtractorConfig):
    model_type = "cage_detector"

    def __init__(self, fixed_watermark=None, **kwargs):
        super().__init__(**kwargs)
        # The fixed signature this detector checks extracted bits against.
        self.fixed_watermark = fixed_watermark