无需训练,用提示词自动检测工业缺陷
Automatic Prompt Generation and Grounding Object Detection for Zero-Shot Image Anomaly Detection
- 用GPT-3生成正常/异常产品描述提示词
- 结合Grounding DINO定位产品,用CLIP比对图像与提示
- 在MVTec-AD和VisA数据集上实现零样本高精度检测
工业产品缺陷检测是关键的质量控制任务。传统人工检测效率低、主观性强且易出错。本文提出一种新型零样本、无需训练的自动化工业图像异常检测方法,基于多模态机器学习流程,包含三个基础模型。首先使用大语言模型GPT-3生成描述正常与异常产品外观的文本提示;随后利用视觉-语言定位模型Grounding DINO定位图像中的产品区域;最后通过零样本图像-文本匹配模型CLIP对比裁剪后的图像块与生成提示,识别异常。在MVTec-AD和VisA两个工业图像数据集上的实验表明,该方法无需模型训练即可高效检测多种缺陷,具备高准确率。所提方法可实现工业制造场景中高效、可扩展且客观的质量控制。
原文摘要 · Abstract (English)
Identifying defects and anomalies in industrial products is a critical quality control task. Traditional manual inspection methods are slow, subjective, and error-prone. In this work, we propose a novel zero-shot training-free approach for automated industrial image anomaly detection using a multimodal machine learning pipeline, consisting of three foundation models. Our method first uses a large language model, i.e., GPT-3. generate text prompts describing the expected appearances of normal and abnormal products. We then use a grounding object detection model, called Grounding DINO, to locate the product in the image. Finally, we compare the cropped product image patches to the generated prompts using a zero-shot image-text matching model, called CLIP, to identify any anomalies. Our experiments on two datasets of industrial product images, namely MVTec-AD and VisA, demonstrate the effectiveness of this method, achieving high accuracy in detecting various types of defects and anomalies without the need for model training. Our proposed model enables efficient, scalable, and objective quality control in industrial manufacturing settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。