公开(公告)号：CN117541868A

公开(公告)日：2024-02-09

申请号：CN202311558413.7

申请日：2023-11-21

Applicant: 复旦大学

Inventor： 戈维峰 ,  孙翊轩 ,  崔晋

IPC: G06V10/764 ,  G06V10/82 ,  G06V10/40 ,  G06V10/74 ,  G06N3/0455 ,  G06N3/09

Abstract: 本申请提供了一种图像分类模型的训练方法、图像分类方法、模型、计算机设备及介质，包括获取包括图像三元组的训练样本，所述图像三元组包括查询图像、与所述查询图像属于相同类别的正图像和与所述查询图像属于不同类别的负图像；提取所述查询图像、所述正图像、所述负图像的向量特征；利用多头交叉注意力层对所述查询图像、所述正图像、所述负图像的向量特征交叉注意处理；利用多头自注意力层增强所述交叉注意力层的输出的表征；基于所述自注意力层的输出计算损失；以及基于所述损失训练所述图像分类模型。

公开(公告)号：CN117541868B

公开(公告)日：2024-12-03

申请号：CN202311558413.7

申请日：2023-11-21

Applicant: 复旦大学

Inventor： 戈维峰 ,  孙翊轩 ,  崔晋

IPC: G06V10/764 ,  G06V10/82 ,  G06V10/40 ,  G06V10/74 ,  G06N3/0455 ,  G06N3/09

Abstract: 本申请提供了一种图像分类模型的训练方法、图像分类方法、模型、计算机设备及介质，包括获取包括图像三元组的训练样本，所述图像三元组包括查询图像、与所述查询图像属于相同类别的正图像和与所述查询图像属于不同类别的负图像；提取所述查询图像、所述正图像、所述负图像的向量特征；利用多头交叉注意力层对所述查询图像、所述正图像、所述负图像的向量特征交叉注意处理；利用多头自注意力层增强所述交叉注意力层的输出的表征；基于所述自注意力层的输出计算损失；以及基于所述损失训练所述图像分类模型。