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