证件识别模型训练方法、装置、存储介质及电子设备与流程
技术特征:
1.一种证件识别模型训练方法,所述证件识别模型包括编码器、解码器及任务引导解释器,所述方法包括:
2.根据权利要求1所述的方法,所述编码器是预训练过的,所述方法还包括:
3.根据权利要求2所述的方法,所述基于自监督训练方式对所述编码器进行预训练,得到预训练过的编码器,包括:
4.根据权利要求1所述的方法,所述确定所述证件图像训练样本相关联的至少一个证件识别任务,包括:
5.根据权利要求1所述的方法,所述确定所述证件图像训练样本相关联的至少一个证件识别任务,包括:
6.根据权利要求1所述的方法,所述获得所述至少一个证件识别任务对应的任务查询向量,包括:
7.根据权利要求6所述的方法,所述关联信息包括关联图像信息,所述方法还包括:
8.根据权利要求7所述的方法,所述对于所述每个证件识别任务,根据该证件识别任务的任务性质,确定所述证件图像训练样本关于该证件识别任务的关联图像信息包括:
9.根据权利要求7所述的方法,所述对于所述每个证件识别任务,根据该证件识别任务的任务性质,确定所述证件图像训练样本关于该证件识别任务的关联图像信息包括:
10.根据权利要求7所述的方法,所述对于所述每个证件识别任务,根据该证件识别任务的任务性质,确定所述证件图像训练样本关于该证件识别任务的关联图像信息包括:
11.根据权利要求1所述的方法,所述任务引导解释器包括多个输出单元,每个证件识别任务对应的输出结果从与该证件识别任务对应的输出单元中输出。
12.根据权利要求11所述的方法,所述任务引导解释器还包括门控制单元及任务引导单元,任务引导单元用于根据所述任务特征信息控制门控制单元的打开和关闭,所述门控制单元用于控制每个证件识别任务对应的输出结果从与该证件识别任务对应的输出单元中输出。
13.根据权利要求12所述的方法,每个输出单元对应一种任务类型,所述门控制单元用于控制每个证件识别任务对应的输出结果从该证件识别任务的任务类型对应的输出单元中输出。
14.根据权利要求13所述的方法,所述任务类型包括分类任务、回归任务、分割任务、检测任务。
15.一种证件识别方法,包括:
16.一种证件识别模型训练装置,所述证件识别模型包括编码器、解码器及任务引导解释器,所述装置包括:
17.一种证件识别装置,包括:
18.一种存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现权利要求1~14或15中任意一项所述方法的步骤。
19.一种电子设备,其特征在于,包括:处理器和存储器;其中,所述存储器存储有计算机程序,所述计算机程序适于由所述处理器加载并执行如权利要求1~14或15中任意一项所述方法的步骤。
20.一种计算机程序产品,其上存储有至少一条指令,其特征在于,所述至少一条指令被处理器执行时实现权利要求1~14或15中任意一项所述方法的步骤。
技术总结
本说明书实施例公开了一种证件识别模型训练方法、装置、存储介质及电子设备,首先获取证件图像训练样本,确定所述证件图像训练样本相关联的至少一个证件识别任务,并获得所述至少一个证件识别任务对应的任务查询向量,之后将所述证件图像训练样本输入所述编码器进行特征提取,得到图像特征信息,将所述任务查询向量及所述图像特征信息输入所述解码器,得到任务特征信息,将所述任务特征信息输入所述任务引导解释器,得到所述任务引导解释器输出的至少一个输出结果,然后根据所述至少一个输出结果,利用损失函数对所述证件识别模型进行训练,得到已训练的证件识别模型。
技术研发人员:陈志军
受保护的技术使用者:蚂蚁区块链科技(上海)有限公司
技术研发日:
技术公布日:2024/11/26
技术研发人员:陈志军
技术所有人:蚂蚁区块链科技(上海)有限公司
备 注:该技术已申请专利,仅供学习研究,如用于商业用途,请联系技术所有人。
声 明 :此信息收集于网络,如果你是此专利的发明人不想本网站收录此信息请联系我们,我们会在第一时间删除
