公开(公告)号：CN120031844A

公开(公告)日：2025-05-23

申请号：CN202510133511.9

申请日：2025-02-06

Applicant: 河南科技大学

Inventor： 霍华 ,  赛格 ,  王航 ,  王利平 ,  赵利群 ,  李彦蓉

IPC: G06T7/00 ,  G06T7/70 ,  G06N3/0464 ,  G06N3/045 ,  G06N3/08 ,  G06V10/80

Abstract: 本发明属于深度学习图像检测技术领域，具体涉及一种改进YOLOv8的安检图像违禁品检测方法和计算机设备。该方法获取安检图像并输入至训练后的违禁品检测模型中，识别出安检图像中违禁品的位置和种类；违禁品检测模型是在YOLOv8上改进得到，改进包括：YOLOv8的主干网络中位于网络最深层的C2f模块和SPPF模块之间增加SimIRB注意力机制模块，SimIRB注意力机制模块是将iRMB模块中的多头注意力机制替换为SimAM注意力机制后的模块。本发明利用SimIRB注意力机制模块可以同时捕获长距离依赖和特征间的全局依赖以增强对物体的特征提取能力，两者相辅相成实现出色的特征提取效果，提升违禁品检测精度。