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