添加对yolox、ssd、faster-rcnn白盒水印提取验证流程

watermark_verify/process/faster-rcnn_pytorch_whitebox_process.py

@@ -0,0 +1,17 @@
+"""
+faster-rcnn基于pytorch框架的白盒水印处理验证流程
+"""
+from watermark_verify.process.general_process_define import WhiteBoxWatermarkProcessDefine
+
+
+class ClassificationProcess(WhiteBoxWatermarkProcessDefine):
+    def __init__(self, model_filename):
+        super(ClassificationProcess, self).__init__(model_filename)
+
+    def process(self) -> bool:
+        """
+        根据流程定义进行处理，并返回模型标签验证结果
+        :return: 模型标签验证结果
+        """
+        verify_result = self.verify_label(start=9, end=13)  # 模型标签检测通过，进行标签验证
+        return verify_result

watermark_verify/process/ssd_pytorch_whitebox_process.py

@@ -0,0 +1,17 @@
+"""
+ssd基于pytorch框架的白盒水印处理验证流程
+"""
+from watermark_verify.process.general_process_define import WhiteBoxWatermarkProcessDefine
+
+
+class ClassificationProcess(WhiteBoxWatermarkProcessDefine):
+    def __init__(self, model_filename):
+        super(ClassificationProcess, self).__init__(model_filename)
+
+    def process(self) -> bool:
+        """
+        根据流程定义进行处理，并返回模型标签验证结果
+        :return: 模型标签验证结果
+        """
+        verify_result = self.verify_label(start=1, end=4)  # 模型标签检测通过，进行标签验证
+        return verify_result

watermark_verify/process/yolox_pytorch_whitebox_process.py

@@ -0,0 +1,17 @@
+"""
+yolox基于pytorch框架的白盒水印处理验证流程
+"""
+from watermark_verify.process.general_process_define import WhiteBoxWatermarkProcessDefine
+
+
+class ClassificationProcess(WhiteBoxWatermarkProcessDefine):
+    def __init__(self, model_filename):
+        super(ClassificationProcess, self).__init__(model_filename)
+
+    def process(self) -> bool:
+        """
+        根据流程定义进行处理，并返回模型标签验证结果
+        :return: 模型标签验证结果
+        """
+        verify_result = self.verify_label(start=60, end=64)  # 模型标签检测通过，进行标签验证
+        return verify_result