期刊文献+

融合ViT卷积神经网络的木板表面缺陷识别 被引量:7

Wood Surface Defect Recognition Based on ViT Convolutional Neural Network
下载PDF
导出
摘要 由于需要通过木板表面缺陷对木板分级,而人工检测存在一定问题。为解决木板表面缺陷识别问题,提出一种融合ViT的卷积神经网络模型,用于提高缺陷识别的准确率。为此,收集裂缝、虫眼、节子和纹理4种木板表面缺陷图片,其中裂缝和虫眼图片数量远少于节子和纹理。为解决模型训练时样本不均衡问题,利用ProGAN对裂缝和虫眼图片进行训练并生成同类型缺陷图片,以增加其数量,使4种图片数量保持平衡,并在实验前对缺陷图片进行数据增强并添加椒盐噪声,整理得到所需图片数据集。基于融合ViT的卷积神经网络模型,利用数据集验证两种不同激活函数的模型,结果表明使用GELU作为激活函数的模型性能更高。并测试不同的transformer深度时模型的性能,得到的模型缺陷识别的最高准确率可达到98.54%。实验结果表明,融合ViT的卷积神经网络模型是可行的,为木板表面缺陷自动检测提供了新思路。 There are some problems in manual detection because it is necessary to grade the wood board through its surface defects.In order to solve the problem of wood surface defect recognition,a convolution neural network model integrating ViT is proposed to improve the accuracy of defect recognition.For this purpose,four kinds of wood surface defect pictures of crack,wormhole,knot and texture are collected,in which the number of crack and wormhole pictures is far less than that of knot and texture.In order to solve the problem of sample imbalance in model training,ProGAN is used to train crack and wormhole pictures and generate pictures of the same type of defects,so as to increase their number and keep the number of four kinds of pictures balanced.Before the experiment,the data of defect images are enhanced and salt and pepper noise is added to sort out the required image data set.Based on the convolutional neural network model fused with ViT,two models with different activation functions are tested by using the data set.It is found that the model using GELU as the activation function has higher performance.The model performance at different transformer depths is tested,and the highest accuracy of model defect identification can reach 98.54%.Experiments show that the convolutional neural network model fused with ViT is feasible,which provides a new idea for the automatic detection of wood surface defects.
作者 郭文龙 刘芳华 吴万毅 李冲 肖鹏 刘朝 GUO Wen-long;LIU Fang-hua;WU Wan-yi;LI Chong;XIAO Peng;LIU Chao(School of Mechanical Engineering,Jiangsu University of Science and Technology,Zhenjiang,Jiangsu 212100,China)
出处 《计算机科学》 CSCD 北大核心 2022年第S02期609-614,共6页 Computer Science
基金 国家自然科学基金青年科学基金(51905228)
关键词 木板表面缺陷 ProGAN VIT 卷积神经网络 深度学习 Wood surface defects ProGAN ViT Convolutional neural network Deep learning
  • 相关文献

参考文献5

二级参考文献50

  • 1Thompson David K, Mandrake Lukas, Gilmore M S, et al. Super-pixel entlmemher dt'tection[ J] . IKEK Transactions On GeoscienceAnd Remote Sensing,2010,48( 11):4023-4033.
  • 2MicuSik Branislav, KoSeck6 Jana. Multi-view superpixel stereo inurban enviromnentsi[ J ]. International Journal of Computer Vision,2010,89(1):106-119.
  • 3Liu Bin, Hu Hao, W ang Huanyu,et al.SuptTpixel-based classifica-tion with an ada[)hve number of classes for pdarimelrit‘ SAH Ima-ges [J].IEEE Transactions On (Geoscience And Heinote Sensing,2013,51(2):907-924.
  • 4Li Shifeng, Lu Hufliuan, Ruan Xiang,et al.Human luxly segmenta-tion hasetl on ciefomiahle models and two-scale superpixel[ J] .Pat-tern Analysis and Applications,2012,15(4) :399-413.
  • 5Fu Keren,Gong Chen,Yang Jie,et al.Superpixel based color con-trast and color distriimtion driven salient object delec*tion[ J ] .Sig-nal Processing:Image Coniniunication,2013,28( 19) : 1448-1463.
  • 6Liu Zhenhua,Jiang Zhentjuan,Zuo Husong. Sluily of fussy cluste-ring of enginering geological envinminenl with GIS [ J ] .Journal ofChina University of Mining & Technology,2003,13(2): 196-200.
  • 7冯少荣,肖文俊.DBSCAN聚类算法的研究与改进[J].中国矿业大学学报,2008,37(1):105-111. 被引量:88
  • 8吴东洋,业宁,沈丽容,张倩倩,赖正文.基于颜色矩的木材缺陷聚类识别[J].江南大学学报(自然科学版),2009,8(5):520-524. 被引量:11
  • 9吴东洋,业宁,苏小青.基于灰度共生矩阵和聚类方法的木材缺陷识别[J].计算机与数字工程,2010,38(11):38-41. 被引量:12
  • 10孙建平,王逢瑚,胡英成.基于声发射和神经网络的木材受力损伤过程检测[J].仪器仪表学报,2011,32(2):342-347. 被引量:22

共引文献40

同被引文献49

引证文献7

相关作者

内容加载中请稍等...

相关机构

内容加载中请稍等...

相关主题

内容加载中请稍等...

浏览历史

内容加载中请稍等...
;
使用帮助 返回顶部