Bladeren bron

修改vgg白盒为3-14层

zhy 4 dagen geleden
bovenliggende
commit
25e568f8fa
2 gewijzigde bestanden met toevoegingen van 3 en 3 verwijderingen
  1. 2 2
      watermark_verify/process/vggnet_whitebox_process.py
  2. 1 1
      watermark_verify/verify_tool_mix.py

+ 2 - 2
watermark_verify/process/vggnet_whitebox_process.py

@@ -8,10 +8,10 @@ class ModelWatermarkProcessor(WhiteBoxWatermarkProcessDefine):
     def __init__(self, model_filename):
         super(ModelWatermarkProcessor, self).__init__(model_filename)
 
-    def process(self) -> bool:
+    def process(self, conv_range=(3, 14)) -> bool:
         """
         根据流程定义进行处理,并返回模型标签验证结果
         :return: 模型标签验证结果
         """
-        verify_result = self.verify_label(scope=(3, 14))  # 模型标签检测通过,进行标签验证
+        verify_result = self.verify_label(scope=conv_range)  # 模型标签检测通过,进行标签验证
         return verify_result

+ 1 - 1
watermark_verify/verify_tool_mix.py

@@ -70,7 +70,7 @@ def label_verification(model_filename: str, framework: str='pytorch', mode: str=
                 message=f"不支持的组合: framework={framework}, mode={mode}, model_type={model_type}"
             )
         if mode == 'whitebox' and model_type == 'vggnet':
-            result = processor_class.process((3, 7))
+            result = processor_class.process((3, 14))
         else:
             result = processor_class.process()