公开(公告)号：CN118397357A

公开(公告)日：2024-07-26

申请号：CN202410549290.9

申请日：2024-05-06

Applicant: 徐州高新区安全应急装备产业技术研究院

Inventor： 雷少华 ,  王梦 ,  李冲 ,  刁秀强 ,  俞啸 ,  董飞 ,  张世源 ,  陈希琳

IPC: G06V10/764 ,  G06V10/82 ,  G06V10/80 ,  G06N3/0464 ,  G06N3/045

Abstract: 本发明涉及改进BIFPN与注意力机制的轻量化小目标检测方法，包括：获取待检测图像，将待检测图像输入改进的小目标检测模型，获取目标检测结果；其中，改进的小目标检测模型利用训练集训练获得，训练集包括VOC数据集和M‑Helmet自制数据集，改进的小目标检测模型通过基于CA注意力机制的小目标特征提取网络模块进行特征提取，基于BIFPN的小目标特征融合强化网络模块进行多尺度特征融合，同时利用EIOU损失函数优化改进的小目标检测模型的目标函数。本发明能够实现较高的目标检测精度，还具有实时性好、模型轻量化的优点。