GAN用于表面缺陷检测

前两个月写的，以为误删了，结果在草稿箱里，再发出来吧，也可以见知乎：https://zhuanlan.zhihu.com/p/50099453

1.前言

深度学习在计算机视觉主流领域已经应用的很成熟，但是在工业领域（某些领域应用也很成熟），比如产品表面缺陷检测，总感觉没有发挥深度学习的强大能力，近几年表面缺陷的 相关研究主要是集中在各种借鉴主流神经网络框架，从CNN到YOLO，SSD，甚至到语义分割的FCN相关论文，通过一些技术，对框架进行轻量化，对缺陷进行分类或检测。不过，逃不出一个基础：一定要有缺陷样本可供训练，而且数量不能太少！当然，也有一些课题组记住稀疏编码、字典学习、稀疏自编码等对表面缺陷进行检测，这类方法很有局限性，主要针对那些有周期性背景纹理的图像，比如丝织品，印刷品等。

很多课题组、工业软件公司都想开发出一些切合实际应用的算法软件，在缺陷检测领域，比较好的公司有：VIDI、Halcon等，听说海康威视也在搞工业产品方便的算法研究。

下面开始说下这篇文章，文章是今年发表的，我不知道前人有没有这样的尝试，作者提出只依据已有的正常表面图像样本，通过一定的技术手段对缺陷样本进行检测，很好的将最近研究火热的GAN应用于框架中，这一年，课题组的老师也一直讨论这种方法的可行性，缺陷的检测要不要有缺陷样本，从稀疏自编码，小样本学习再到计算机视觉研究热点之一的零样本学习，得出结论：大多数工业产品表面缺陷检测是需要缺陷样本或者人为制作的缺陷样本，论文虽然是没有直接使用生产线上的缺陷样本，但是通过算法人为的产生了缺陷样本，并很好的融合和GAN在图像修复领域的强大能力，整个框架的设计很巧妙。

论文下载链接：A Surface Defect Detection Method Based on Positive Samples

文章思路：论文的整体思路就是GAN在图像修复和重建方便具有很强大的能力，通过人为的去在正常样本上“随意”添加一些缺陷，训练阶段让GAN去学习一个可以修复这些缺陷区域的网络，检测阶段时，输入一个真实缺陷样本，训练好的G