Browse Source

重命名模型水印处理流程类

liyan 4 months ago
parent
commit
1ecdc904b9

+ 2 - 2
watermark_verify/process/classification_all_whitebox_process.py

@@ -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:
         """

+ 2 - 2
watermark_verify/process/classification_pytorch_blackbox_process.py

@@ -11,9 +11,9 @@ from watermark_verify.process.general_process_define import BlackBoxWatermarkPro
 import onnxruntime as ort
 
 
-class ClassificationProcess(BlackBoxWatermarkProcessDefine):
+class ModelWatermarkProcessor(BlackBoxWatermarkProcessDefine):
     def __init__(self, model_filename):
-        super(ClassificationProcess, self).__init__(model_filename)
+        super(ModelWatermarkProcessor, self).__init__(model_filename)
 
     def process(self) -> bool:
         """

+ 2 - 2
watermark_verify/process/classification_tensorflow_blackbox_process.py

@@ -11,9 +11,9 @@ from watermark_verify.process.general_process_define import BlackBoxWatermarkPro
 import onnxruntime as ort
 
 
-class ClassificationProcess(BlackBoxWatermarkProcessDefine):
+class ModelWatermarkProcessor(BlackBoxWatermarkProcessDefine):
     def __init__(self, model_filename):
-        super(ClassificationProcess, self).__init__(model_filename)
+        super(ModelWatermarkProcessor, self).__init__(model_filename)
 
     def process(self) -> bool:
         """

+ 2 - 2
watermark_verify/process/googlenet_all_whitebox_process.py

@@ -4,9 +4,9 @@ GoogleNet基于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:
         """