[发明专利]一种模型训练方法、装置、电子设备及存储介质

权利要求书

1.一种模型训练方法，其特征在于，所述方法包括：

将训练图像样本输入第一模型，得到所述训练图像样本的第一特征；所述训练图像样本表征含有边框信息的页面设计图；所述第一特征表征所述训练图像样本中边框信息的框架样式特征；

将框架组件样本输入第二模型，得到所述框架组件样本的第二特征；所述框架组件样本是基于对所述训练图像样本中的边框信息转换得到的张量数据；所述第二特征表征所述框架组件样本对应的框架样式特征；

基于所述第一特征与所述第二特征的相似度，确定设定的损失函数对应的损失值；

根据所述损失值更新所述第一模型和/或所述第二模型的权重参数。

2.根据权利要求1所述的方法，其特征在于，所述第一模型和所述第二模型共享输出层。

3.根据权利要求1所述的方法，其特征在于，所述第一模型和所述第二模型基于yolo模型构建。

4.一种检索方法，其特征在于，所述方法包括：

将检索对象输入至与所述检索对象的类型对应的特征提取模型，得到所述检索对象对应的框架样式特征；所述特征提取模型包括基于权利要求1至3任一项所述的模型训练方法训练得到的第一模型或第二模型；

将所述框架样式特征输入至第一数据库进行匹配，得到所述检索对象对应的检索结果；其中，

所述检索对象的类型包括图片或者框架组件；所述第一数据库为与检索结果的类型相关的数据库；所述检索结果的类型包括图片或者框架组件。

5.根据权利要求4所述的方法，其特征在于，在所述检索对象为图片的情况下，所述检索对象的类型对应的特征提取模型为所述第一模型；所述将所述框架样式特征输入至第一数据库进行匹配，得到所述检索对象对应的检索结果，包括：

确定所述框架样式特征与所述第一数据库中存储的每个第一特征之间的相似度；所述第一数据库存储有至少两张图片和所述至少两张图片中每张图片对应的第一特征；所述第一特征表征框架样式特征；

基于所述相似度确定所述检索结果。

6.根据权利要求4所述的方法，其特征在于，在所述检索对象为图片的情况下，所述检索对象的类型对应的特征提取模型为所述第一模型；所述将所述框架样式特征输入至第一数据库进行匹配，得到所述检索对象对应的检索结果，包括：

确定所述框架样式特征与所述第一数据库中存储的每个第二特征之间的相似度；所述第一数据库存储有至少两个框架组件和所述至少两个框架组件中每个框架组件对应的第二特征；所述第二特征表征框架样式特征；

基于所述相似度确定所述检索结果。

7.根据权利要求4所述的方法，其特征在于，在所述检索对象为框架组件的情况下，所述检索对象的类型对应的特征提取模型为所述第二模型；所述将所述框架样式特征输入至第一数据库进行匹配，得到所述检索对象对应的检索结果，包括：

确定所述框架样式特征与第一数据库中存储的每个第二特征之间的相似度；所述第一数据库存储有至少两个框架组件和所述至少两个框架组件中每个框架组件对应的第二特征；所述第二特征表征框架样式特征；

基于所述相似度确定所述检索结果。

8.根据权利要求4所述的方法，其特征在于，在所述检索对象为框架组件的情况下，所述检索对象的类型对应的特征提取模型为所述第二模型；所述将所述框架样式特征输入至第一数据库进行匹配，得到所述检索对象对应的检索结果，包括：

确定所述框架样式特征与所述第一数据库中存储的每个第一特征之间的相似度；所述第一数据库存储有至少两张图片和所述至少两张图片中每张图片对应的第一特征；所述第一特征表征框架样式特征；

基于所述相似度确定所述检索结果。