LED灯带缺陷检测模型的训练方法、装置、计算机可读存储介质及LED灯带缺陷检测方法
Abstract:
本发明实施例提供一种LED灯带缺陷检测模型的训练方法、装置、计算机可读存储介质及LED灯带缺陷检测方法,训练方法包括:提供LED灯带训练集;从LED灯带训练集中选取出K个典型正类样本,并存入记忆池;获取各个正类样本对应的伪异常样本;以及将剩余正类样本和伪异常样本作为输入图像进行模型训练:采用特征提取网络提取输入图像在各个预定维度的高级特征信息;获得输入图像和记忆样本之间的差异信息,并确定最佳差异信息;获得串联信息;基于串联信息获得融合特征图;以及获取空间注意力图,并将各个空间注意力图通过跳跃连接流向解码器输出预测图像。本实施例能提高模型的鲁棒性,方便不同数据集上进行扩展。
Patent Agency Ranking
0/0