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

摘要

本发明公开了一种基于注意力机制的图像检测方法，可以通过注意力机制确定图片中感兴趣的区域，包括：采集图像，获取需要测试的图像数据集；将图像分成独立不重复的验证集和测试集；对验证集和测试集中的图像进行特征提取，获取需要的特征信息；基于Darknet53网络模型增加由通道注意力模块和空间注意力模块构成的SCSE模块，得到测试图像的模型；将验证集内的图像特征作为输入模型参数；将测试集内的图像特征作为输入模型参数；将测试集内图像的特征输入，获取相应的测试结果。本实验可以提高图片检测的精度，同时也能够提升检测的效率，提高资源的利用率。