汤寓麟, 边少锋, 翟国君, 刘敏, 张卫东. 侧扫声纳检测沉船目标的改进YOLOv5法[J]. 武汉大学学报 ( 信息科学版). DOI: 10.13203/j.whugis20210353
引用本文: 汤寓麟, 边少锋, 翟国君, 刘敏, 张卫东. 侧扫声纳检测沉船目标的改进YOLOv5法[J]. 武汉大学学报 ( 信息科学版). DOI: 10.13203/j.whugis20210353
TANG Yulin, BIAN Shaofeng, ZHAI Guojun, LIU Min, ZHANG Weidong. Improved YOLOv5 Method for Detecting Shipwreck Target with Side-scan Sonar[J]. Geomatics and Information Science of Wuhan University. DOI: 10.13203/j.whugis20210353
Citation: TANG Yulin, BIAN Shaofeng, ZHAI Guojun, LIU Min, ZHANG Weidong. Improved YOLOv5 Method for Detecting Shipwreck Target with Side-scan Sonar[J]. Geomatics and Information Science of Wuhan University. DOI: 10.13203/j.whugis20210353

侧扫声纳检测沉船目标的改进YOLOv5法

Improved YOLOv5 Method for Detecting Shipwreck Target with Side-scan Sonar

  • 摘要: 针对基于YOLOv3模型的侧扫声纳沉船目标检测方法存在小目标漏警率高、模型权重大、检测速度未能满足实时性需求等问题,引入YOLOv5算法并根据数据集特点,提出基于YOLOv5模型的侧扫声纳海底沉船目标检测方法。在YOLOv5基础框架下尝试YOLOv5a、YOLOv5b、YOLOv5c、YOLOv5d、YOLOv5s、YOLOv5m、YOLOv5l和YOLOv5x八种不同深度和宽度的模型结构,并选择最优的结构,使用GA+K(Genetic Algorithm+K-mean)算法优化检测框,通过CIOU_Loss对损失函数进行改进。实验结果表明,改进的YOLOv5a模型在AP_0.5和AP_0.5-0.9较原始模型提高了0.3%和0.6%,较YOLOv3模型有了全面大幅提升,其中AP_0.5和AP_0.5-0.9分别提高了4.2%和6.1%,检测速度达到426帧/秒,提升了几乎一倍,更加益于实际应用和工程部署。

     

    Abstract: In view of the problems of high miss-alarm rate of small targets, heavy model weight, and detection speed that fails to meet real-time requirements in side-scan sonar shipwreck detection method based on the YOLOv3 model. The paper introduces the YOLOv5 algorithm and proposes a model based on YOLOv5 according to the characteristics of the side-scan sonar shipwreck dateset.Try YOLOv5a, YOLOv5b, YOLOv5c, YOLOv5d, YOLOv5s, YOLOv5m, YOLOv5l and YOLOv5x under the basic framework of YOLOv5 with eight different depth and width model structures.Choose the best structure use GA+K (Genetic Algorithm+K-mean) algorithm to optimize the detection frame, and improve the loss function through CIOU_Loss. The experimental results show that the improved YOLOv5a model is 0.3% and 0.6% higher than the original model in AP_0.5 and AP_0.5-0.9, and has a substantial improvement compared with the YOLOv3 model, in which AP_0.5 and AP_0.5-0.9 are improved by 4.2% and 6.1% and the the detection speed reaches 426 frames per second, which is almost doubled that of YOLOv3, which is more conducive to practical applications and engineering deployment.

     

/

返回文章
返回