一种基于图像扩散的工业产品缺陷检测方法

本发明公开了一种基于图像扩散的工业产品缺陷检测方法，在训练阶段，对输入图片进行图像扩散，并将扩散后的图片输入到多输出扩散模型，将多输出扩散模型的输出与扩散后的图片进行对比，计算损失值并更新模型参数；在测试阶段，将工业产品图片输入扩散模型得到中间预测结果，再将中间预测结果作为扩散模型输入循环若干次，得到最终输出，其中通过动态步数预测模块预测工业产品图片的循环次数；本发明可以用于对工业产品图片进行缺陷检测，同时完成检测和分割任务。