轻量化改进网络的海天线检测模型

摘要：

海天线检测在无人艇自动驾驶与海上目标识别领域具有重要作用.针对传统算法在各类场景下鲁棒性差的问题,提出一种基于目标识别算法的海天线检测模型YOLOv5-SH.该模型利用ShuffleNetV2 网络单元搭建特征提取网络,替换YOLOv5 模型的骨干网络,提取图像中海天线区域,再利用Hough变换的方法在其中进行线性检测,识别海天线.在自建数据集上的对比实验结果表明,YOLOv5-SH模型能够在各类场景下实时准确检测海天线.

关键词：机器视觉、海天线、目标检测、轻量化网络

所属期刊栏目：48

分类号：TP391(计算技术、计算机技术)

资助基金：国家自然科学基金61501486

在线出版日期：2023-07-06（万方平台首次上网日期，不代表论文的发表时间）

页数：共8页

页码：18-24,32