基于YOLOv3-Tiny改进的船舶目标检测研究

IMPROVED SHIP TARGET DETECTION BASED ON YOLOV3-TINY

  • 摘要: 针对船舶实时性检测中出现的检测精度低、漏检问题,改进一种基于YOLOv3-Tiny的船舶目标检测算法。通过引入深度可分离卷积作为主干网络,提高通道数量,减少模型的参数量和运算量;采用H-Swish和Leaky ReLU激活函数改进卷积结构,提取更多特征信息;利用GIOU(Generalized Intersection Over Union)损失优化边界框,突显目标区域重合度,提高精度。在混合船舶数据集上检测结果表明,改进后YOLOv3-Tiny的检测精度为83.40%,较原算法提高5.33百分点,召回率和检测速度也均优于原算法,适用于船舶实时性检测。

     

    Abstract: Aiming at the problems of low detection accuracy and missed detection in real-time detection of ships, this paper improves a ship target detection algorithm based on YOLOV3-Tiny. By introducing depth wise separable convolution as the backbone network, the number of channels was increased and the number of parameters and computation of the model were reduced. The H-Swish and Leaky ReLU activation functions were used to improve the convolution structure in order to extract more feature information. GIOU loss was used to optimize the bounding box to highlight the coincidence degree of the target area and improve the accuracy. The detection results on the mixed ship data set show that the detection accuracy of the improved YOLOv3-Tiny is 83.40%, which is 5.33 percentage points higher than the original algorithm. Its recall rate and detection speed are also better than the original algorithm, which is suitable for real-time ship detection.

     

/

返回文章
返回