计算机与现代化

• 算法设计与分析 • 上一篇    下一篇

基于自编码器图像重构的织物瑕疵检测算法

  

  1. (福州大学数学与计算机科学学院,福建福州350116)
  • 收稿日期:2018-04-24 出版日期:2019-01-30 发布日期:2019-01-30
  • 作者简介:欧庆芳(1992-),男,福建莆田人,硕士研究生,研究方向:计算机视觉,机器学习,E-mail: ouqf123@163.com; 谢伙生(1964-),男,福建三明人,副教授,硕士,研究方向:数据挖掘,图像处理,虚拟现实,机器学习。

Fabric Defect Detection Based on Image Reconstruction with Auto-encoder

  1. (College of Mathematics and Computer Science, Fuzhou University, Fuzhou 350116, China)
  • Received:2018-04-24 Online:2019-01-30 Published:2019-01-30

摘要: 针对含周期图案织物瑕疵检测通常要计算周期,而这又不适应纯色织物,本文提出适应2类织物的检测方法。首先设定检测块大小,按其在无瑕疵图像中随机提取图像块,并训练自编码器。然后将待检图像按设定大小分块,用自编码器重构,并计算重构前后的均方误差。最后对计算结果进行异常值检测,均方误差值偏大的为瑕疵块。实验表明,本文算法适应2类织物,容易实现,检测效果较好。

关键词: 分块, 自编码器, 均方误差, 瑕疵检测

Abstract: The fabric defect detection with periodic pattern generally needs to calculate period, which is not suitable for pure texture fabric. In this paper, the detection method for two kinds of fabric is proposed. Firstly, the size of detect block is set, the image block is extracted randomly from non-defect images, and then the auto-encoder is trained. Secondly, according to the size, the test images are divided into several blocks, and reconstructed with the auto-encoder, and then the mean square errors are computed between the reconstructed result and original data. Lastly, outliers of computing results of test images are detected. With rather large mean square errors value, the blocks are defect blocks. The experiments show that the proposed method is suitable for two kinds of fabric, the process is more practical and the detection results are better.

Key words:  partitioning, auto-encoder, mean square error, defect detection

中图分类号: