A method for workpiece surface small-defect detection based on CutMix and YOLOv3; [基于CutMix和YOLOv3的工件表面小缺陷识别方法]

被引:0
|
作者
Xing J. [1 ]
Jia M. [1 ]
Xu F. [1 ]
Hu J. [1 ]
机构
[1] School of Mechanical Engineering, Southeast University, Nanjing
基金
中国国家自然科学基金;
关键词
Deep convolutional neural network; Defect detection; Image recognition; Machine vision;
D O I
10.3969/j.issn.1003-7985.2021.02.002
中图分类号
学科分类号
摘要
Surface small defects are often missed and incorrectly detected due to their small quantity and unapparent visual features. A method named CSYOLOv3, which is based on CutMix and YOLOv3, is proposed to solve such a problem. First, a four-image CutMix method is used to increase the small-defect quantity, and the process is dynamically adjusted based on the beta distribution. Then, the classic YOLOv3 is improved to detect small defects accurately. The shallow and large feature maps are split, and several of them are merged with the feature maps of the predicted branch to preserve the shallow features. The loss function of YOLOv3 is optimized and weighted to improve the attention to small defects. Finally, this method is used to detect 512 × 512 pixel images under RTX 2060Ti GPU, which can reach the speed of 14.09 frame/s, and the mAP is 71.80%, which is 5%-10% higher than that of other methods. For small defects below 64 × 64 pixels, the mAP of the method reaches 64.15%, which is 14% higher than that of YOLOv3-GIoU. The surface defects of the workpiece can be effectively detected by the proposed method, and the performance in detecting small defects is significantly improved. © 2021, Editorial Department of Journal of Southeast University. All right reserved.
引用
收藏
页码:128 / 136
页数:8
相关论文
共 50 条
  • [1] 基于CutMix和YOLOv3的工件表面小缺陷识别方法(英文)
    邢俊杰
    贾民平
    许飞云
    胡建中
    Journal of Southeast University(English Edition), 2021, 37 (02) : 128 - 136
  • [2] A SURFACE DEFECT DETECTION METHOD OF STEEL PLATE BASED ON YOLOV3
    Ouyang, G. Z.
    Zhang, W. Y.
    METALURGIJA, 2023, 62 (01): : 61 - 64
  • [3] 基于改进YOLOv3的车辆识别方法
    王永顺
    贾文杰
    王晨飞
    宋慧
    激光与光电子学进展, 2021, 58 (16) : 240 - 247
  • [4] A metal surface defect detection method based on improved YOLOv3
    Wu, Kai
    Wang, Yuhong
    2023 35TH CHINESE CONTROL AND DECISION CONFERENCE, CCDC, 2023, : 4887 - 4892
  • [5] 基于YOLOv3的稻穗识别方法研究
    彭祺
    屠礼芬
    程瑞敏
    吴涛
    湖北工程学院学报, 2021, 41 (06) : 56 - 60
  • [6] 基于改进YOLOv3的数字识别方法
    陈悦哲
    孔令云
    李杰
    张仪
    王许鹏
    科学技术创新, 2020, (28) : 28 - 29
  • [7] 基于改进YOLOv3的玉米病害识别方法
    张继成
    黄向党
    中国农机化学报, 2024, (07) : 269 - 275
  • [8] 基于改进YOLOv3的手势实时识别方法
    张强
    张勇
    刘芝国
    周文军
    刘佳慧
    计算机工程, 2020, 46 (03) : 237 - 245+253
  • [9] 基于改进YOLOv3的目标识别方法
    陈正斌
    叶东毅
    朱彩霞
    廖建坤
    计算机系统应用, 2020, 29 (01) : 49 - 58
  • [10] 基于改进YOLOv3的输电线路缺陷识别方法
    陈嘉琛
    俞曜辰
    陈中
    韩卫
    南方电网技术, 2021, 15 (01) : 114 - 120