文献类型：期刊文献

中文题名：基于迁移学习和集成神经网络的皮革缺陷检测

英文题名：Ensemble leather defect detection based on transfer learning

作者：侯灿阳[1];朱北辰[1];吴清[1]

机构：[1]华东理工大学机械与动力工程学院,上海201424

年份：2023

卷号：49

期号：5

起止页码：70

中文期刊名：东华大学学报（自然科学版）

外文期刊名：Journal of Donghua University(Natural Science)

收录：CSTPCD;;北大核心:【北大核心2020】；

基金：国家自然科学基金面上项目(52075176)。

语种：中文

中文关键词：缺陷检测;皮革;迁移学习;集成学习;测试时增强;K折交叉验证

外文关键词：defect detection;leather;transfer learning;ensemble learning;test-time augmentation;K-fold cross validation test

摘要：皮革缺陷检测在工业生产中具有重要作用。人工检测效率低且主观性强,因此自动化皮革缺陷检测具有重要意义。为克服人工检测和传统目标检测方法的诸多缺点、实现自动化缺陷检测,基于YOLOv5算法提出了一种新型皮革缺陷检测方法。该方法利用迁移学习、集成学习和测试时增强(test-time augmentation, TTA)提高了算法的泛化能力,采用K折交叉验证的方法避免过拟合。试验结果表明:基于深度学习的目标检测方法相比传统方法具有显著优势。相比其他深度学习方法,所提出的检测框架能将每帧检测时间缩短至24 ms内,同时检测精度达96.58%,对于工业皮革生产具有实用价值。
Leather defect detection has an unignorable importance in industry.Considering that manual detection is time-consuming and to objective,automated detection is necessary.A new detection method based on YOLOv5 algorithm is proposed to overcome shortcomings of manual detection and traditional object detection methods and realize automatic defect detection.The proposed method combines strategy of transfer learning,ensemble learning and test-time augmentation to enhance ability of generalization.K-fold cross validation test is used to prevent over fitting.Experimental results show that detection methods based on deep learning excel traditional ones,and our proposed method has a precision up to 96.58%while outperforming other deep learning algorithms in speed.Each frame is predicted within 24 ms,which can be valuable to industrial production.