公开(公告)号：CN117830732A

公开(公告)日：2024-04-05

申请号：CN202410008519.8

申请日：2024-01-02

Applicant: 上海工程技术大学

Inventor： 罗光圣 ,  李烨恒 ,  方志军 ,  苏前敏

IPC: G06V10/764 ,  G06V20/60 ,  G06V10/80 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/048 ,  G06N3/08

Abstract: 本发明公开了基于YOLOv7模型的Logo检测方法和装置,所述方法包括：构建采用YOLOv7模型作为基础框架的Logo检测模型，骨干网络包括第一特征提取模块和依次连接的三组第二特征提取模块；第二特征提取模块包括CSMP模块和ELAN模块；通过第一组第二特征提取模块的CSMP模块对最大池化后和平均池化后的第一特征图分别进行基于通道注意力机制和基于空间注意力机制的特征提取，得到通道注意力特征图和空间注意力特征图，拼接通道注意力特征图和空间注意力特征图，得到增强特征图，通过ELAN模块处理增强特征图，得到第二特征图；通过Logo检测模型的头部网络对三组第二特征图进行处理，得到Logo图像样本的Logo检测结果。采用本方法能够对不同的Logo图像进行检测，提高Logo检测的精度。