乐清市盐盆街道纬五路222号乐清加速器B幢203

+86 15669738183

图像分类方法、系统、介质及电子设备

本发明提供一种图像分类方法、系统、介质及电子设备,包括构建图像分类模型,图像分类模型包括文本表征提取器、视觉表征提取器、可学习度量模块;构建扩展类别标签;基于视觉表征提取器提取支援图像的支援视觉表征,提取查询图像的查询视觉表征;基于可学习度量模块计算文本表征和支援视觉表征的第一相似度矩阵,调整可学习度量模块的参数;基于可学习度量模块计算文本表征和查询视觉表征的第二相似度矩阵,并调整文本表征提取器、视觉表征提取器和可学习度量模块的参数,以基于训练好的图像分类模型实现待分类图像的分类。本发明的图像分类方法、系统、介质及电子设备通过图像、文本双模态对比学习的方式进行图像分类模型的训练,提升了准确性。

More Posts

Send Us A Message