Jian-qi Li, Yincong Liang, Rui Du, Jingying Wan, Bin-fang Cao, Hui Liu
{"title":"Lightweight defect detection method of punched nickel-plated steel strip based on GhostNet","authors":"Jian-qi Li, Yincong Liang, Rui Du, Jingying Wan, Bin-fang Cao, Hui Liu","doi":"10.1109/prmvia58252.2023.00017","DOIUrl":null,"url":null,"abstract":"Aiming at the problem that the defects generated in the production and transportation of punched nickel-plated steel strips are not easy to be detected by deep learning methods, a lightweight, low-redundancy, and high-precision detection method is proposed in this paper. Firstly, a feature extraction network based on GhostNet is constructed, which reduces the amount of computation and feature redundancy while ensuring accuracy. Then the ECA module is applied to the detection head to perform weighted fusion of the features of different channels for better differentiation. Finally, the YOLO detection head is used for multi-scale detection. In the experiment, the mAP of 84.86% was obtained by this method, which proves that this method can be applied to the actual steel strip defect: detection.","PeriodicalId":221346,"journal":{"name":"2023 International Conference on Pattern Recognition, Machine Vision and Intelligent Algorithms (PRMVIA)","volume":null,"pages":null},"PeriodicalIF":0.0000,"publicationDate":"2023-03-01","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":null,"platform":"Semanticscholar","paperid":null,"PeriodicalName":"2023 International Conference on Pattern Recognition, Machine Vision and Intelligent Algorithms (PRMVIA)","FirstCategoryId":"1085","ListUrlMain":"https://doi.org/10.1109/prmvia58252.2023.00017","RegionNum":0,"RegionCategory":null,"ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"","JCRName":"","Score":null,"Total":0}
引用次数: 0
Abstract
Aiming at the problem that the defects generated in the production and transportation of punched nickel-plated steel strips are not easy to be detected by deep learning methods, a lightweight, low-redundancy, and high-precision detection method is proposed in this paper. Firstly, a feature extraction network based on GhostNet is constructed, which reduces the amount of computation and feature redundancy while ensuring accuracy. Then the ECA module is applied to the detection head to perform weighted fusion of the features of different channels for better differentiation. Finally, the YOLO detection head is used for multi-scale detection. In the experiment, the mAP of 84.86% was obtained by this method, which proves that this method can be applied to the actual steel strip defect: detection.
京公网安备 11010802042870号
京ICP备2023020795号-1
分享
求助内容：
应助结果提醒方式：
扫码关注我们
