一种基于注意力机制的图像检测方法

    公开(公告)号:CN113344138A

    公开(公告)日:2021-09-03

    申请号:CN202110765391.6

    申请日:2021-07-06

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

Patent Agency Ranking