CADN: A weakly supervised learning-based category-aware object detection network for surface defect detection
文献类型:期刊论文
作者 | Jiabin, Zhang1,2; Hu, Su2; Wei, Zou1,2; Xinyi, Gong2; Zhengtao, Zhang2; Fei, Shen2 |
刊名 | Pattern Recognition |
出版日期 | 2020-08 |
期号 | 109页码:1-9 |
关键词 | Weakly supervised learning Automated surface inspection Defect detection Knowledge distillation |
英文摘要 | Large-scale data with human annotations is of crucial importance for training deep convolutional neural network (DCNN) to ensure stable and reliable performance. However, accurate annotations, such as bounding box and pixel-level annotations, demand expensive labeling efforts, which has prevented wide application of DCNN in industries. Focusing on the problem of surface defect detection, this paper proposes a weakly supervised learning method named Category-Aware object Detection network (CADN) to tackle the dilemma. CADN is trained with image tag annotations only and performs image classification and defect localization simultaneously. The weakly supervised learning is achieved by extracting category-aware spatial information in a classification pipeline. CADN could be equipped with either a lighter or a larger backbone network as the feature extractor resulting in better real-time performance or higher accuracy. To address the two conflicting objectives simultaneously, both of which are significant concerns in industrial applications, knowledge distillation strategy is adopted to force the learned features of a lighter CADN to mimic that of a larger CADN. Accordingly, the accuracy of the lighter CADN is improved while high real-time performance is maintained. The proposed approach is verified on our own defect dataset as well as on an open-source defect dataset. As demonstrated, satisfied performance is achieved by the proposed method, which could meet industrial requirements completely. Meanwhile, the method minimizes human efforts involved in image labelling, thus promoting the applications of DCNN in industries. |
语种 | 英语 |
源URL | [http://ir.ia.ac.cn/handle/173211/44779] |
专题 | 精密感知与控制研究中心_精密感知与控制 |
通讯作者 | Hu, Su |
作者单位 | 1.University of Chinese Academy of Sciences 2.Institute of Automation, Chinese Academy of Sciences |
推荐引用方式 GB/T 7714 | Jiabin, Zhang,Hu, Su,Wei, Zou,et al. CADN: A weakly supervised learning-based category-aware object detection network for surface defect detection[J]. Pattern Recognition,2020(109):1-9. |
APA | Jiabin, Zhang,Hu, Su,Wei, Zou,Xinyi, Gong,Zhengtao, Zhang,&Fei, Shen.(2020).CADN: A weakly supervised learning-based category-aware object detection network for surface defect detection.Pattern Recognition(109),1-9. |
MLA | Jiabin, Zhang,et al."CADN: A weakly supervised learning-based category-aware object detection network for surface defect detection".Pattern Recognition .109(2020):1-9. |
入库方式: OAI收割
来源:自动化研究所
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。