公开(公告)号：CN118521765A

公开(公告)日：2024-08-20

申请号：CN202410479884.7

申请日：2024-04-22

Applicant: 中电科新防务技术有限公司 ,  华北水利水电大学

Inventor： 韩振宇 ,  刘辉 ,  余国河 ,  曹连海 ,  周学军 ,  王博达 ,  任朝栋 ,  张继民 ,  李葛爽 ,  杨世纪

IPC: G06V10/25 ,  G06V10/80 ,  G06V10/764 ,  G06V10/44 ,  G06N3/0464 ,  G06N3/045

Abstract: 一种用于水面漂浮物检测的边缘融合YOLOv5算法，解决已有目标检测算法在速度和准确性上较差的问题。其包括包括：特征提取、特征融合和多分类器模块；包括以下步骤：S1、特征提取：利用快速边缘聚焦卷积模块生成V1、V2和V3三幅特征图，其特别凸显了边缘信息，再对特征图V2和b进行相加，作为新的特征图V2b输出；S2、特征融合：在不同尺度特征图基础上进行空间金字塔池化，以增强尺度不变性，引入路径聚合网络和注意力机制提高特征复杂度，再通过上采样和下采样进一步凸显边缘细节；S3、多分类器模块进行特征点判断：设计三种独特的Yolo Head分类器，对特征融合模块输出的融合特征进行特征点判断，以实现高精度分类识别。