一种基于YOLOv8的轻量级水下小目标检测方法
摘要:
本发明提供了一种基于YOLOv8的轻量级水下小目标检测方法,包括:步骤1:建立水下小目标的图像集;步骤2:对水下小目标的图像集进行预处理;步骤3:以YOLOv8为基准搭建水下小目标检测网络模型并对模型进行优化;步骤4:利用预处理后的水下小目标的图像集对优化后的水下小目标检测网络模型进行训练,对训练后的水下小目标检测网络模型进行预测和评估,基于预测和评估后的水下小目标检测网络模型获取水下小目标检测结果。本发明可满足于真实环境中嵌入式设备部署目标检测器的轻量化需求,同时端到端的目标检测器具有优良的鲁棒性能与实时性能。
0/0