一种基于无偏嵌入零样本学习的电芯缺陷分类方法

摘要

一种基于无偏嵌入零样本学习的电芯缺陷分类方法，包括：1)从通过高清摄像头采集的电芯图片中，采用滑动窗口的方式获取合适大小的包含缺陷的图像块，将较为常见的缺陷类别分配为带标签的源数据，将少见的缺陷类别分配为无标签的目标数据；2)使用有标注的源类数据和未标注的目标类数据来端对端地训练准全监督学习网络模型QFSL；3)测试阶段，对于输入的图像块，通过视觉嵌入子网络得到它的视觉嵌入，接着利用视觉‑语义衔接子网络，完成从视觉嵌入到语义嵌入的映射，然后通过内积计算得到投影得到的视觉嵌入和语义嵌入的得分，最后送入Softmax分类器产生了所有类别的预测概率，将概率最高的那个类别作为分类结果。