在审

图像分类及其模型训练方法、装置、电子设备、存储介质

张保林、何浩、段文平、黄勇波、伍海银、苏军、杨超、曾鑫、仁义、冯长、肖顺利
赛力斯汽车有限公司

摘要

本申请涉及人工智能技术领域,提供了一种图像分类及其模型训练方法、装置、电子设备、存储介质,图像分类模型的训练方法包括:获取多个训练图像和多个训练文本;对多个所述训练图像中每个所述训练图像编码得到多个训练图像特征向量;对多个所述训练文本中每个所述训练文本编码得到多个训练文本特征向量;将每个所述训练图像特征向量分别与多个所述训练文本特征向量中每个训练文本特征向量嵌入同一空间并得到一个正样本对向量和多个负样本对向量;分别计算一个正样本对向量之间和多个负样本对向量之间的差异,得到多个损失差异;基于多个损失差异进行模型训练得到所述图像分类模型。应用本方案,能够解决现有技术中存在的图像分类效率过低的问题。

暂无引用专利