在审

基于特征聚合损失的长尾视觉识别方法、装置及电子设备

李硕豪、李子豪、杨佳鑫、张军、雷军、王翔汉、黄魁华、陈超、刘忠
中国人民解放军国防科技大学

摘要

本发明提供了一种基于特征聚合损失的长尾视觉识别方法、装置及电子设备,包括从长尾图像数据集中获取训练样本,基于训练样本训练长尾视觉识别模型;训练过程中,首先对长尾视觉识别模型进行正则化处理,然后依次对长尾视觉识别模型添加影响因子使得训练样本在特征空间分别进行分离操作和聚合操作;再将分离操作的影响因子和聚合操作的影响因子进行聚合得到特征聚合损失,基于特征聚合损失对长尾视觉识别模型进行训练,得到训练后的长尾视觉识别模型;采用训练后的长尾视觉识别模型对输入的待识别图像进行识别。通过对样本在特征空间中的收缩、分离和聚合操作,增强了难样本的识别能力;特征在空间的类分布边界更加清晰,从而提高了识别准确率。

暂无引用专利