首页|基于自动机器学习的织物瑕疵检测方法

基于自动机器学习的织物瑕疵检测方法

扫码查看
在解决许多实际的计算机视觉问题时,卷积神经网络(CNN)已经展现了比传统机器学习更强大的能力。但在解决一个具体的问题时,需要针对特定数据集,在掌握该领域具体知识情况下,才能设计出比较有效的CNN结构。而在这个构造过程中,需要花费大量的计算资源和时间。论文提出一种自动机器学习方法,即基于随机漂移粒子群优化(RDPSO)算法自动搜索得到最优的单分类深度支持向量数据描述(DeepSVDD)网络结构的方法(称为rdpsoCNN),应用于织物瑕疵检测。实验证明,与一些经典的CNN结构相比,文中所提出的rdpsoCNN方法优化的模型具有更好的瑕疵检测能力。
Fabric Defect Detection Method Based on Automatic Machine Learning
While solving many practical problems,the methods based on convolutional neural network(CNN)have shown better performance than traditional machine learning methods.However,as to a specific problem and data set,an efficient CNN model can only be designed based on knowing domain knowledge completely.Usually it will take plenty of computing resource and time to finish the designing procedure.This paper proposes a new automatic machine learning method(named rdpsoCNN)for fabric defect detection,the random drift particle swarm optimization algorithm(RDPSO)is used to automatically find an optimal CNN structure of one-class deep support vector data description.The experimental results show that the proposed rdpsoCNN can find good CNN structures that achieve better detection performance compared to the state-of-art designs.

automatic machine learningconvolutional neural networkone-class deep support vector data descriptionran-dom drift particle swarm optimizationdefect detection

罗海驰、张家玮、李岳阳

展开 >

江南大学物联网工程学院 无锡 214122

江南大学人工智能与计算机学院 无锡 214122

自动机器学习 卷积神经网络 单分类深度支持向量数据描述 随机漂移粒子群优化算法 瑕疵检测

2024

计算机与数字工程
中国船舶重工集团公司第七0九研究所

计算机与数字工程

CSTPCD
影响因子:0.355
ISSN:1672-9722
年,卷(期):2024.52(11)