基于改进YOLOv8n模型的河道目标识别方法及装置

本发明公开了一种基于改进YOLOv8n模型的河道目标识别方法及装置，属于目标识别技术领域，包括：S1、构建河道环境数据集，包括：收集与河道周围环境信息相关的图片；对图片进行标注，得到河道环境数据集；S2、在YOLOv8n基础模型上搭建改进YOLOv8n的河道目标识别网络，河道目标识别网络包括小目标检测层、SPD‑Conv空间深度转换卷积模块和ECA注意力机制，得到改进YOLOv8n模型；S3、将河道环境数据集导入改进YOLOv8n模型中进行训练、验证和测试，得到改进后的YOLOv8n河道目标识别模型；S4、利用所述YOLOv8n河道目标识别模型对目标图片进行河道目标识别。