交叉口(航空)
卷积神经网络
目标检测
计算机科学
数据集
人工智能
探测器
集合(抽象数据类型)
点(几何)
功能(生物学)
像素
模式识别(心理学)
高斯分布
数据挖掘
计算机视觉
数学
工程类
电信
物理
几何学
量子力学
进化生物学
生物
程序设计语言
航空航天工程
作者
Yaxin Tu,Zhigang Ling,Siyu Guo,He Wen
标识
DOI:10.1109/tim.2020.3024431
摘要
Although one-stage detectors, for example, YOLOv3, SSD, and RefineDet, based on the convolutional neural networks (CNNs) have been widely used in object detection, their localization accuracy of the defects is still poor and the problem of repeated detection still occurs. This article proposes a surface defects detection method based on improved YOLOv3 for sawn lumbers. First, a Gaussian YOLOv3 is used to estimate the coordinates and the localization uncertainty of the prediction box. Then, the complete intersection over union (CIoU) loss function is adopted to replace the intersection over union (IoU) loss for non-maximum suppression (NMS), because it considers the overlap area, the center point, and the aspect ratio, and thus can reduce the repeated detection. Finally, this article establishes two data sets, that is, the rubber lumber data set containing four kinds of defects and the pine lumber data set containing two kinds of defects. In total, there are 16 633 images and 6705 annotated images in these two data sets. The effectiveness of the proposed method has been validated by experiments. Our data sets are available at https://github.com/WenHe-Hnu/sawn-lumber-dataset.
科研通智能强力驱动
Strongly Powered by AbleSci AI