-
公开(公告)号:CN118115795A
公开(公告)日:2024-05-31
申请号:CN202410242631.8
申请日:2024-03-04
Applicant: 哈尔滨工业大学(深圳)(哈尔滨工业大学深圳科技创新研究院)
IPC: G06V10/764 , G06V10/774 , G06T7/00 , G06V10/82 , G06N3/0464
Abstract: 本申请提供了电子元器件分类方法、设备及存储介质,该方法包括:获取已标注的图像和未标注的图像,并将已标注的图像划分为训练集和测试集;基于训练集和测试集训练多个不同类型的神经网络模型,得到多个电子元器件分类模型,并将其划分为主模型和副模型;分别通过主模型和副模型在未标注的图像中筛选,筛选得到第一样本集合和第二样本集合;基于第一样本集合和第二样本集合,得到目标样本集合;将目标样本集合加入训练集中,重新训练主模型和副模型,检验重新训练后的主模型的分类准确率是否满足预设条件,若满足预设条件,则通过重新训练后的主模型对电子元器件进行分类。本申请能够提高综合区分良品以及不同类型的不良品的能力。
-
公开(公告)号:CN118397350A
公开(公告)日:2024-07-26
申请号:CN202410511838.0
申请日:2024-04-26
Applicant: 哈尔滨工业大学(深圳)(哈尔滨工业大学深圳科技创新研究院)
IPC: G06V10/764 , G06V10/74 , G06N20/00
Abstract: 本发明提供一种基于提示词学习的图像分类方法、装置、设备及介质。该方法包括:获取待分类图像、预设的多个标签和每个标签对应的多个初始提示词;分别将每个标签和对应的多个初始提示词输入至预设分类模型的文本编码器中,得到文本编码器输出的每个标签对应的第一文本向量;利用预设通道因子,对每个标签对应的第一文本向量进行调整,获得每个标签对应的第二文本向量;根据待分类图像对应的第一图像向量,以及每个标签对应的第二文本向量,确定待分类图像与每个标签的相似度;基于待分类图像与每个标签的相似度,对待分类图像进行分类。本发明能够实现快速准确地进行图像分类。
-