-
公开(公告)号:CN115965828A
公开(公告)日:2023-04-14
申请号:CN202110529942.9
申请日:2021-05-14
Applicant: 华为云计算技术有限公司
IPC: G06V10/774 , G06V10/82 , G06V10/764 , G06V10/77 , G06N3/0464
Abstract: 本申请实施例提供了一种图像分类方法和电子设备。本申请实施例提供的技术方案中,通过编码器对有监督训练集和无监督测试集进行特征提取,得到所述有监督训练集在隐层空间的第一低维特征和所述无监督测试集在所述隐层空间的第二低维特征;根据所述第一低维特征得到每个类别的中心原型向量;根据所述中心原型向量和所述第二低维特征得到所述第二低维特征与所述中心原型向量的相似性;根据所述相似性挑选出正样本;通过将所述正样本与所述无监督测试集进行对比学习,得到图像分类模型。本申请实施例通过将对比学习引入半监督学习领域,能够利用少量的标签对数据进行分类,并达到良好的分类效果。