公开(公告)号：CN119722582A

公开(公告)日：2025-03-28

申请号：CN202411684273.2

申请日：2024-11-22

Applicant: 珠海格力电器股份有限公司

Inventor： 王春龙 ,  郭琛 ,  刘志昌 ,  胡宗群 ,  程江昊

IPC: G06T7/00 ,  G06V10/764 ,  G06V10/774 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/084

Abstract: 本发明实施例提供了一种器件缺陷检测方法、装置、电子设备及可读存储介质，所述方法包括：获取器件样本图像；将器件样本图像输入待训练的器件缺陷检测模型，得到器件缺陷检测模型输出的不同时刻对应的预测类别和预测概率；根据预测类别和预测概率确定器件缺陷检测模型的预测变化；根据预测变化更新器件样本图像对应的权重；根据更新权重的器件样本图像训练器件缺陷检测模型；当器件缺陷检测模型满足预设收敛条件时，得到训练完成的目标器件缺陷检测模型。本发明实施例可以在正常器件样本图像和缺陷器件样本图像的数量不同的情况下，对于学习难度高的少数类别，动态分配比较高的权重，提高器件缺陷检测模型在少数类别上的识别性能。