修改水印检测工具类名

watermark_verify/process/ssd_pytorch_blackbox_process.py

@@ -12,9 +12,9 @@ from watermark_verify.tools import parse_qrcode_label_file
 from watermark_verify.tools.evaluate_tool import calculate_ciou
 
 
-class DetectionProcess(BlackBoxWatermarkProcessDefine):
+class ModelWatermarkProcessor(BlackBoxWatermarkProcessDefine):
     def __init__(self, model_filename):
-        super(DetectionProcess, self).__init__(model_filename)
+        super(ModelWatermarkProcessor, self).__init__(model_filename)
 
     def process(self) -> bool:
         # 获取权重文件，使用触发集进行模型推理, 将推理结果与触发集预先二维码保存位置进行比对，在误差范围内则进行下一步，否则返回False

watermark_verify/process/ssd_pytorch_whitebox_process.py

@@ -4,9 +4,9 @@ ssd基于pytorch框架的白盒水印处理验证流程
 from watermark_verify.process.general_process_define import WhiteBoxWatermarkProcessDefine
 
 
-class DetectionProcess(WhiteBoxWatermarkProcessDefine):
+class ModelWatermarkProcessor(WhiteBoxWatermarkProcessDefine):
     def __init__(self, model_filename):
-        super(DetectionProcess, self).__init__(model_filename)
+        super(ModelWatermarkProcessor, self).__init__(model_filename)
 
     def process(self) -> bool:
         """

watermark_verify/process/yolox_pytorch_blackbox_process.py

@@ -9,9 +9,9 @@ from watermark_verify.tools import parse_qrcode_label_file
 from watermark_verify.tools.evaluate_tool import calculate_ciou
 
 
-class DetectionProcess(BlackBoxWatermarkProcessDefine):
+class ModelWatermarkProcessor(BlackBoxWatermarkProcessDefine):
     def __init__(self, model_filename):
-        super(DetectionProcess, self).__init__(model_filename)
+        super(ModelWatermarkProcessor, self).__init__(model_filename)
 
     def process(self) -> bool:
         """

watermark_verify/process/yolox_pytorch_whitebox_process.py

@@ -4,9 +4,9 @@ yolox基于pytorch框架的白盒水印处理验证流程
 from watermark_verify.process.general_process_define import WhiteBoxWatermarkProcessDefine
 
 
-class DetectionProcess(WhiteBoxWatermarkProcessDefine):
+class ModelWatermarkProcessor(WhiteBoxWatermarkProcessDefine):
     def __init__(self, model_filename):
-        super(DetectionProcess, self).__init__(model_filename)
+        super(ModelWatermarkProcessor, self).__init__(model_filename)
 
     def process(self) -> bool:
         """

watermark_verify/verify_tool.py

@@ -1,4 +1,4 @@
-from watermark_verify.process import yolox_pytorch_blackbox_process
+from watermark_verify.process.yolox_pytorch_blackbox_process import ModelWatermarkProcessor
 
 
 def label_verification(model_filename: str) -> bool:
@@ -8,7 +8,7 @@ def label_verification(model_filename: str) -> bool:
     :return: 模型标签验证结果
     """
     # 初始化模型水印检测器
-    model_detector = yolox_pytorch_blackbox_process.DetectionProcess(model_filename)
+    model_detector = ModelWatermarkProcessor(model_filename)
     result = model_detector.process()  # 获取模型水印检测结果
     return result