|
@@ -4,9 +4,9 @@ AlexNet、VGG16、ResNet基于pytorch、tensorflow、keras框架的白盒水印
|
|
|
from watermark_verify.process.general_process_define import WhiteBoxWatermarkProcessDefine
|
|
|
|
|
|
|
|
|
-class ClassificationProcess(WhiteBoxWatermarkProcessDefine):
|
|
|
+class ModelWatermarkProcessor(WhiteBoxWatermarkProcessDefine):
|
|
|
def __init__(self, model_filename):
|
|
|
- super(ClassificationProcess, self).__init__(model_filename)
|
|
|
+ super(ModelWatermarkProcessor, self).__init__(model_filename)
|
|
|
|
|
|
def process(self) -> bool:
|
|
|
"""
|