[发明专利]图像识别模型的训练方法、装置、电子设备和存储介质

权利要求书

1.一种图像识别模型的训练方法，包括：

获取训练数据集，其中，所述训练数据集中包括第一图像集、第二图像集及所述第一图像集中每个第一图像对应的标注结果，其中，所述第二图像集中的每个第二图像为对应的第一图像经过第一变换后生成的；

将任一第一图像及对应的第二图像分别输入初始孪生网络中的第一编码器及第二编码器，以获取所述任一第一图像对应的第一特征图及所述对应的第二图像对应的第二特征图；

将所述第一特征图及所述第二特征图分别输入所述初始孪生网络中的第一解码器及第二解码器，以获取第一预测结果及第二预测结果；

根据所述第二特征图、所述第一特征图、所述第一预测结果、所述第二预测结果及所述任一第一图像对应的标注结果，对所述初始孪生网络进行修正，以获取第一图像识别网络及第二图像识别网络。

2.如权利要求1所述的方法，其中，所述对所述初始孪生网络进行修正，以获取第一图像识别网络及第二图像识别网络，包括：

将所述第二特征图进行第二变换，以获取第三特征图，其中，所述第二变换与所述第一变换互为逆变换；

确定所述第三特征图与所述第一特征图间的第一差异；

确定所述第一预测结果与所述任一第一图像对应的标注结果间的第二差异；

确定所述第二预测结果与所述任一第一图像对应的标注结果间的第三差异；

基于所述第一差异、所述第二差异及所述第三差异，对所述初始孪生网络进行修正，以获取第一图像识别网络及第二图像识别网络。

3.如权利要求2所述的方法，其中，所述基于所述第一差异、所述第二差异及所述第三差异，对所述初始孪生网络进行修正，以获取第一图像识别网络及第二图像识别网络，包括：

将所述第一差异、所述第二差异及所述第三差异进行融合，以确定损失值；

在所述损失值大于阈值的情况下，基于所述损失值对所述初始孪生网络进行修正，以获取第一图像识别网络及第二图像识别网络。

4.如权利要求1所述的方法，其中，所述将所述任一第一图像及所述对应的第二图像分别输入初始孪生网络中的第一编码器及第二编码器，包括：

根据所述任一第一图像中每个第一像素点的第一坐标值，确定所述第一图像对应的第一位置向量；

根据所述对应的第二图像中每个第二像素点的第二坐标值，确定所述第二图像对应的第二位置向量；

将所述任一第一图像及所述第一位置向量输入所述第一编码器，将所述对应的第二图像及所述第二位置向量输入所述第二编码器。

5.如权利要求1-4任一所述的方法，其中，所述获取第一预测结果及第二预测结果，包括：

获取所述任一第一图像分别属于各类图像的第一概率，及属于各类图像时对应的第一尺寸偏差值；

获取所述对应的第二图像分别属于各类图像的第二概率，及属于各类图像时对应的第二尺寸偏差值；

根据各个所述第一概率，确定所述任一第一图像所属的第一目标类型；

根据所述任一第一图像属于所述第一目标类型时对应的第一尺寸偏差值及所述第一目标类型对应的第一参考尺寸，确定所述任一第一图像中物体的第一尺寸；

根据各个所述第二概率，确定所述对应的第二图像所属的第二目标类型；

根据所述对应的第二图像属于所述第二目标类型时对应的第二尺寸偏差值及所述第二目标类型对应的第二参考尺寸，确定所述对应的第二图像中物体的第二尺寸。

6.如权利要求5所述的方法，其中，所述标注结果中包括每个第一图像中物体的尺寸及类型，所述方法还包括：

根据所述每个第一图像中物体的类型，确定每类图像对应的子图像集，其中，每个子图集中的各个第一图像中的物体类型相同；

根据每个子图像集中的各个第一图像中物体的尺寸，确定每类图像对应的参考尺寸。

7.如权利要求1-4所述的方法，其中，所述标注结果中包括以下至少一项：物体二维检测框的尺寸，物体的尺寸，物体的三维中心点，物体的朝向角以及物体所属的类型。