基于YOLOv8的水下目标检测方法、装置及存储介质

    公开(公告)号:CN118470512A

    公开(公告)日:2024-08-09

    申请号:CN202410615966.X

    申请日:2024-05-17

    摘要: 本发明涉及计算机视觉领域,特别涉及一种基于YOLOv8的水下目标检测方法、装置及存储介质;所述方法包括获取水下图像;将所述水下图像输入到水下目标检测模型中,得到所述水下目标检测模型输出的水下目标识别结果;所述水下目标检测模型基于改进YOLOv8模型构建而成;所述水下目标检测模型包括特征提取网络Backbone模块、特征融合网络Neck模块和特征识别网络Head模块;所述Backbone模块中的C2f模块替换成可变形卷积模块C2f‑DCNv4;所述Neck模块采用加权双向特征金字塔网络,所述Neck模块嵌入有SimAM注意力机制;本发明采用了基于改进YOLOv8模型的水下目标检测模型,能够降低水下目标检测任务的计算复杂度,提高水下目标检测的精准度,能够更加满足水下目标检测任务的需求。