[发明专利]一种基于注意力机制的图片特征检测方法

权利要求书

1.一种基于注意力机制的图片特征检测方法，其特征在于，该识别方法包括以下几个步骤：

A、改进的可对图片中人体进行识别的YOLO模型；

B、基于注意力机制的特征检测算法。

2.根据权利要求1所述的基于注意力机制的图片特征检测方法，其特征在于，所述步骤A中改进的可对图片中人体进行识别的YOLO模型，具体过程如下：

A1、将传统YOLO模型的输出设计成为输出是注意力区域的灰度图矩阵。

A2、设计自适应裁剪算法,避免因为yolo识别的误差早晨注意力区域图像的损失。注意力区域需要裁剪的宽和高分别为：W＝w+w/30,H＝h+h/30，其中W，H分别为最终要裁剪的宽和高，w，h为yolo模型输出的宽和高。

3.根据权利要求1所述的基于注意力机制的图片特征检测方法，其特征在于，所述步骤B中基于注意力机制的特征检测算法，具体过程如下：

B1、利用自行采集的图片，训练针对人体识别的YOLO模型；

B2、利用训练的YOLO模型确定注意力区域，即图片中人体(除头部)；

B3、将注意力区域转换为灰度图，并输出其灰度图矩阵；

B4、利用SIFT算法检测注意力区域内的特征。