王鹏九, 龚俊斌, 罗威, 黄骁, 郭俊杰. 基于改进Deformable DETR的水面目标检测[J]. 中国舰船研究. DOI: 10.19693/j.issn.1673-3185.03645
引用本文: 王鹏九, 龚俊斌, 罗威, 黄骁, 郭俊杰. 基于改进Deformable DETR的水面目标检测[J]. 中国舰船研究. DOI: 10.19693/j.issn.1673-3185.03645
Detection of Water Surface Targets Based on Improved Deformable DETR[J]. Chinese Journal of Ship Research. DOI: 10.19693/j.issn.1673-3185.03645
Citation: Detection of Water Surface Targets Based on Improved Deformable DETR[J]. Chinese Journal of Ship Research. DOI: 10.19693/j.issn.1673-3185.03645

基于改进Deformable DETR的水面目标检测

Detection of Water Surface Targets Based on Improved Deformable DETR

  • 摘要: 摘 要:【目的】旨在提出一种基于改进Deformable DETR的目标检测算法实现对水面目标的智能识别。【方法】使用轻量化的MobileNetV3替换Deformable DETR原有特征提取网络并引入CBAM注意力机制模块,实现更加高效鲁棒的水面目标检测。通过在自构建的水面目标数据集和公开数据集ABOships开展消融实验以及横向对比试验验证了改进算法的有效性。【结果】在自构建数据集和ABOships两个数据集上的消融实验结果证明,改进算法模型相较原算法模型参数量及大小减少至三分之一,模型推理速度分别提升52.0%和82.7%,mAP0.5:0.95分别提升2.4%和7.5%,训练耗时分别为原算法的41.7%和51.9%。在ABOships数据集上进行的不同算法性能的对比测试结果进一步证明了提出的改进算法在推理速度和检测精度综合性能的优越性。【结论】本文构建了一个新的水面目标数据集,通过对Deformable DETR算法进行改进,大幅提升算法模型推理和训练速度的同时提高了检测准确率,展现了DETR类算法在水面目标检测领域的有效性和应用潜力。

     

    Abstract: Abstract:ObjectiveThis study aims to propose an improved object detection algorithm based on Deformable DETR for intelligent recognition of water surface targets.Methods By substituting the original feature extraction network of Deformable DETR with a lightweight MobileNetV3 and introducing the CBAM attention mechanism module, the study achieves more efficient and robust detection of water surface targets. The effectiveness of the improved algorithm was verified through ablation experiments and horizontal comparative trials conducted on a self-constructed surface water target dataset and the publicly available ABOships dataset.ResultsAblation experiments conducted on both the self-constructed dataset and the ABOships dataset have demonstrated that the improved algorithm significantly reduces the model's parameter count and size to one-third of the original model. In terms of inference speed, there is an increase of 52.0% and 82.7% respectively on these datasets, while the mean Average Precision (mAP) at 0.5:0.95 has improved by 2.4% and 7.5%, and the training time has been reduced to merely 41.7% and 51.9% of the original algorithm, respectively. Further comparative tests of different algorithms conducted on the ABOships dataset underscore the superior performance of the proposed improved algorithm in both inference speed and detection accuracy. ConclusionsThis study has constructed a new water surface target dataset and made significant improvements to the Deformable DETR algorithm. These improvements have not only greatly increased the inference and training speed of the model but also enhanced its detection accuracy. This demonstrates the effectiveness and potential of DETR-class algorithms in the field of water surface target detection.

     

/

返回文章
返回