.png){kind=icon}
返回
DiffDD: A surface defect detection framework with diffusion probabilistic model
.png){kind=icon}
PRE
AI
DOI：10.1016/j.aei.2024.102637.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
Automated defect detection is critical to improve product quality, reduce costs, and enhance productivity in the manufacturing process. More recently, diffusion -based generative models have entered a new realm of vision tasks by offering high -quality and diverse images. However, the diffusion -based defect detection model fails to be researched, and generative models typically require postprocessing modules to achieve the segmentation of defective regions. Furthermore, generic diffusion models struggle to reconstruct defective regions while preserving the semantic information of the inspection images. To address these issues, we propose a Diffusion -based Defect Detection (DiffDD) framework, comprising a pre -trained backbone (PvTv2) and diffusion probabilistic model (DPM), for surface defect detection. The backbone is employed to extract multi -scale feature maps from input images supervised by pixel -wise labels. These pyramid maps interact with DPM to guide the diffusion process, refining the semantic information of the reconstructed image. Additionally, a semantic restoration (SR) module is used to restore the output of the backbone for loss computation with the pixel -wise label. A conditional image prior method is implemented to constrain the randomness of DPM, aiding the model in generating segmentation masks corresponding to the inspection images. Experimental results on NEU_SEG and MT demonstrate that our model outperforms state-of-the-art (SOTA) methods by a significant margin, indicating the generalization and effectiveness of the proposed DiffDD framework.
Keyword:
Diffusion model
Surface defect detection
Semantic segmentation
Generative model
期刊
IF:
9.9
论文数:
4.1K
被引数:
1.7W
机构
引用论文
.png){kind=icon}
Inflexibility of mental planning: A characteristic disorder with prefrontal lobe lesions?心理计划的僵化: 前额叶病变的特征性障碍？
PREAI
分享收藏Sensorless Control of Z Source Inverter fed BLDC Motor Drive by FOC - DTC Hybrid Control Strategy Using Fuzzy Logic Controller采用模糊逻辑控制器的foc-dtc混合控制策略的Z源逆变器馈电BLDC电机驱动的无传感器控制
PREAI
分享收藏Mixed supervision for surface-defect detection: From weakly to fully supervised learning表面缺陷检测的混合监督: 从弱监督学习到完全监督学习
OAAI
OAAI分享收藏
查看更多内容