基于扩散模型的无监督学习药片缺陷检测方法及相关装置

本发明提供了一种基于扩散模型的无监督学习药片缺陷检测方法及相关装置，属于药片缺陷检测技术领域。本发明方法先利用辅助重建网络提取原始的药片图像的语义信息，再利用提取的原始的药片图像的语义信息辅助扩散模型去噪网络对得到的潜在空间向量进行重建，得到重建的药片图像；对原始的药片图像和重建的药片图像分别进行特征提取，得到原始的药片图像特征和重建的药片图像特征；利用原始的药片图像特征和重建的药片图像特征，计算不同尺度的欧式距离与余弦距离；将不同尺度的余弦距离与欧式距离进行加权，得到异常分数；根据异常分数对药片缺陷进行检测，得到药片缺陷检测结果。本发明解决了重建效果不佳、检测准确率不高的问题。