detector_benchmark.watermark.auto_watermark

Attributes

WATERMARK_MAPPING_NAMES

Classes

AutoWatermark

This is a generic watermark class that will be instantiated as one of the watermark classes of the library when

Functions

watermark_name_from_alg_name(name)

Get the watermark class name from the algorithm name.

Module Contents

detector_benchmark.watermark.auto_watermark.WATERMARK_MAPPING_NAMES

detector_benchmark.watermark.auto_watermark.watermark_name_from_alg_name(name)

Get the watermark class name from the algorithm name.

class detector_benchmark.watermark.auto_watermark.AutoWatermark

This is a generic watermark class that will be instantiated as one of the watermark classes of the library when created with the [AutoWatermark.load] class method.

This class cannot be instantiated directly using __init__() (throws an error).

load(algorithm_config=None, gen_model=None, model_config=None, *args, **kwargs)

Load the watermark algorithm instance based on the algorithm name.