一种基于YOLOv8的海上船舶目标快速识别方法

本发明公开了一种基于YOLOv8的海上船舶目标快速识别方法，应用于目标识别技术领域。包括以下步骤：采集数字运载器中船舶航行水域的图像样本后对图像样本进行清洗及标注，将图像样本进行数据增强，得到可用于训练和验证的数据集，基于YOLOv8深度神经网络构建目标识别网络模型，通过设置学习率、批处理大小等超参数，并使用预训练模型进行迁移学习，应用非极大值抑制算法去除重复检测框并使用置信度阈值进行结果过滤来对模型的推理结果进行后处理，完成对船舶目标的预测，对训练后的模型神经网络进行稀疏化剪枝训练，对剪枝后的模型进行校准和量化。本发明在最大化保留模型检测精度的前提下减少了设备运行时的内存消耗，同时加快了模型的检测速度。