一种基于改进YOLOv8s算法的航道船舶及航标检测方法
申请号:CN202410770897
申请日期:2024-06-14
公开号:CN118799622B
公开日期:2025-05-02
类型:发明专利
摘要
一种基于改进YOLOv8s算法的航道船舶及航标检测方法,包括以下步骤:步骤S1:对捕捉到的目标航道的船舶及航标数据集进行筛选整理;步骤S2:经过翻转、裁剪、亮度调整、模糊处理等图像预处理步骤,得到完整数据集;步骤S3:使用Labelimg对数据集进行标注,并按一定的比例划分为训练集、验证集、测试集;步骤S4:构建GCE‑YOLOv8s网络;步骤S5:将所获得的数据集放入GCE‑YOLOv8s模型中,结合GCE‑YOLOv8s模型对获取的船舶及航标图像进行识别,并使用召回率、平均精度均值、以及每秒传输帧数作为评价指标;步骤S6、使用已训练好的模型对待检测图像进行图像预测,精确识别。
技术关键词
航标
检测头
上采样
空间金字塔
船舶
网络
注意力机制
数据
训练集
图像
标注工具
卷积模块
算法
亮度
指标
精度
图片
分支