公开(公告)号：CN117649544A

公开(公告)日：2024-03-05

申请号：CN202311392497.1

申请日：2023-10-24

Applicant: 海南大学

Inventor： 张卫东 ,  冯威翔 ,  贾泽华 ,  薛珊 ,  张云飞 ,  张义博 ,  张安民 ,  郭东生 ,  曹刚 ,  任佳 ,  邹勇华 ,  张文波

IPC: G06V10/762 ,  G06V10/774 ,  G06V10/82 ,  G06V10/24 ,  G06V20/10

Abstract: 本发明涉及一种轻量化水上目标检测方法、装置及介质，其中方法包括以下步骤：获取水上目标的图像数据，构建水上目标数据集；构建基于改进YOLOv7的目标检测模型，采用轻量化的线性瓶颈逆残差模块重构特征提取模块，并引入坐标注意力机制替换SE模块，同时，使用SPD结合非跨步卷积层的形式替代YOLOv7中的下采样模块；使用聚类算法对水上目标数据集进行聚类，将聚类后的数据集分配给不同尺度检测头，对目标检测模型进行训练；对训练后的目标检测模型进行结构重参数化；将采集到的图像输入结构重参数化后的目标检测模型，得到目标的位置和类别置信度信息。与现有技术相比，本发明能够在有限计算资源下实现快速精准的水上目标检测。