公开(公告)号：CN118212403A

公开(公告)日：2024-06-18

申请号：CN202410448486.9

申请日：2024-04-15

Applicant: 郑州轻工业大学

Inventor： 李玉华 ,  张春雨 ,  张梦月 ,  许彦如 ,  黄万伟 ,  李璞 ,  张王卫

IPC: G06V10/25 ,  G06V10/40 ,  G06V10/44 ,  G06V10/80 ,  G06N3/0464 ,  G06N3/047 ,  G06N3/08

Abstract: 本发明提出了一种基于空间上下文特征金字塔的小目标检测方法，其步骤如下：在空间金字塔上增加一条聚合路线构建空间上下文金字塔块模块；构建动态感知上采样模块；在基础网络Yolov5s的骨干网络之后以及颈部网络3个输出的特征图之后融合空间上下文金字塔块模块；在基础网络Yolov5s的颈部网络的前两个卷积层之后添加动态感知上采样模块，生成三个尺寸的特征图输入检测头网络，得到小目标检测模型；采用EIoU损失函数对小目标检测模型进行训练，得到训练后的小目标检测模型；将待检测的图像输入训练后的小目标检测模型，得到小目标检测结果。本发明充分提取了小目标物体的上下文特征信息，在较大的感受野中将空间信息与语义信息充分聚合，更加全面的区分小目标物体与背景信息，提高了小目标检测的检测精度。