首页 | 官方网站   微博 | 高级检索  
     

基于集成学习的铸件缺陷识别方法
引用本文:林庭武,罗坤宇,常建涛,程涵,冯世杰.基于集成学习的铸件缺陷识别方法[J].电子机械工程,2020,36(5):55-61.
作者姓名:林庭武  罗坤宇  常建涛  程涵  冯世杰
作者单位:中兴通讯股份有限公司;西安电子科技大学
基金项目:陕西省重点研发计划项目(2020ZDLGY07-08);国家自然科学基金资助项目(51875432, 51505357)
摘    要:针对铸件图像噪声多和对比度不足引起的缺陷识别困难的问题,文中提出了一种基于集成学习的铸件缺陷识别方法。首先,该方法采用灰度变换法、双边滤波以及自适应图像分割法对铸件图像进行预处理。然后,通过提取方向梯度直方图(Histogram of Oriented Gradients, HOG)特征、不变矩特征和局部二值模式(Local Binary Pattern, LBP)纹理特征构建全信息特征集,并结合支持向量机递归特征消除(Support Vector Machine-Recursive Feature Elimination, SVM-RFE)算法筛选铸件缺陷敏感特征。最后,利用Adaboost-RF(Adaptive Boosting-Random Forest)方法构建铸件缺陷识别模型。对比实验结果表明,该模型不仅可以有效提取缺陷敏感特征,而且相较于其他分类器具有更好的分类性能和泛化能力。

关 键 词:缺陷识别  特征提取  SVM-RFE  Adaboost-RF  集成学习

Casting Defect Recognition Method Based on Ensemble Learning
LIN Tingwu,LUO Kunyu,CHANG Jiantao,CHENG Han,FENG Shijie.Casting Defect Recognition Method Based on Ensemble Learning[J].Electro-Mechanical Engineering,2020,36(5):55-61.
Authors:LIN Tingwu  LUO Kunyu  CHANG Jiantao  CHENG Han  FENG Shijie
Affiliation:ZTE Corporation;Xidian University
Abstract:Aiming at the difficulty in identifying defects caused by excessive noise and insufficient contrast in casting images, a method for identifying defects in castings based on ensemble learning is proposed in this paper. At first, grayscale transformation, bilateral filtering and adaptive image segmentation are employed to preprocess the casting image. Then, the HOG (Histogram of Oriented Gradients) feature, moment invariant feature, and LBP (Local Binary Pattern) texture feature are extracted to construct the full-information feature set. Meanwhile, the SVM-RFE (Support Vector Machine-Recursive Feature Elimination) algorithm is utilized to select sensitive features. In the end, the Adaboost-RF (Adaptive Boosting-Random Forest) method is used to recognize the casting defect. The results of the comparison experiment show that this method can effectively extract the sensitive features from the full-information feature set. Moreover, it has better classification performance and generalization ability than other classifiers.
Keywords:defect recognition  feature extraction  SVM-RFE  Adaboost-RF  ensemble learning
点击此处可从《电子机械工程》浏览原始摘要信息
点击此处可从《电子机械工程》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司    京ICP备09084417号-23

京公网安备 11010802026262号