一种新型轻量化水下小目标检测方法
摘要:
一种新型轻量化水下小目标检测方法,它涉及一种水下小目标检测方法。本发明为了解决虽然基于CNN的检测网络能大幅提高传统检测效率,但在小目标检测和轻量化方面仍有不足的问题。本发明具体包括步骤1、收集水下小目标数据集或采用公开水下数据集;步骤2、对水下数据集进行预处理;步骤3、将预处理后的数据集输入新型轻量化水下小目标检测网络,输入图像经过轻量化多尺度特征提取主干网络提取多尺度特征信息,再经过双塔PAN特征融合网络将深层的语义信息和底层的位置信息进行多尺度融合,最后轻量化耦合检测头分别对像素大小为4×4、8×8和16×16以上的物体进行检测和分类;步骤4、对网络模型进行评估分析。本发明属于目标检测技术领域。
0/0