[发明专利]图像检测方法、装置、电子设备及介质

权利要求书

1.一种图像检测方法，其特征在于，所述方法包括：

利用特征金字塔网络对待检测图像进行不同尺度的特征提取，得到多个不同尺度的精细特征图；

利用预构建的特征蒸馏子网络对所述多个不同尺度的精细特征图进行特征蒸馏，生成待提取特征图；

利用预构建的区域提取网络提取所述待提取特征图中的拟检测区域，并生成所述拟检测区域的预测边界框；

对所述拟检测区域进行分类识别得到识别结果，并对所述预测边界框进行回归定位得到定位结果，根据所述识别结果和所述定位结果在所述待检测图像中进行标记，得到所述待检测图像的检测结果；

其中，所述利用预构建的特征蒸馏子网络对所述多个不同尺度的精细特征图进行特征蒸馏，生成待提取特征图，包括：将所述多个不同尺度的精细特征图按照尺度从小到大进行融合，并加入注意力机制，得到融合特征图；将所述融合特征图通过卷积操作进行特征重构，得到多个不同尺度的新特征图；将所述多个不同尺度的新特征图进行融合，生成待提取特征图。

2.如权利要求1所述的图像检测方法，其特征在于，所述利用特征金字塔网络对待检测图像进行不同尺度的特征提取，得到多个不同尺度的精细特征图，包括：

利用所述特征金字塔网络中的从底向上的通路对所述待检测图像进行逐级下采样，得到多个不同尺度的粗糙特征图；

利用所述特征金字塔网络中的从顶向下的通路和横向连接将所述多个不同尺度的粗糙特征图进行上采样和融合，得到多个不同尺度的精细特征图。

3.如权利要求1所述的图像检测方法，其特征在于，所述利用预构建的区域提取网络提取所述待提取特征图中的拟检测区域，并生成所述拟检测区域的预测边界框，包括：

利用所述待提取特征图采用均衡采样法生成特征块，根据所述特征块将所述待提取特征图划分为不同的图像块，得到拟检测区域；

设置所述拟检测区域中的特征的锚点，根据所述锚点的位置信息生成预测边界框。

4.如权利要求1所述的图像检测方法，其特征在于，所述对所述拟检测区域进行分类识别得到识别结果，并对所述预测边界框进行回归定位得到定位结果，根据所述识别结果和所述定位结果在所述待检测图像中进行标记，得到所述待检测图像的检测结果，包括：

利用预设的激活函数对所述拟检测区域进行分类识别；

在识别出所述拟检测区域为目标区域时，利用所述激活函数对所述目标区域的预测边界框进行回归定位，得到定位结果，其中所述定位结果包括所述目标区域的位置信息；

根据所述位置信息在所述待检测图像中将所述目标区域进行标记，得到所述待检测图像的检测结果。

5.一种图像检测装置，其特征在于，所述装置包括：

特征提取模块，用于利用特征金字塔网络对待检测图像进行不同尺度的特征提取，得到多个不同尺度的精细特征图；

特征蒸馏模块，用于利用预构建的特征蒸馏子网络对所述多个不同尺度的精细特征图进行特征蒸馏，生成待提取特征图；

区域提取模块，用于利用预构建的区域提取网络提取所述待提取特征图中的拟检测区域，并生成所述拟检测区域的预测边界框；

识别定位模块，用于对所述拟检测区域进行分类识别得到识别结果，并对所述预测边界框进行回归定位得到定位结果，根据所述识别结果和所述定位结果在所述待检测图像中进行标记，得到所述待检测图像的检测结果；

其中，在利用预构建的特征蒸馏子网络对所述多个不同尺度的精细特征图进行特征蒸馏，生成待提取特征图时，所述特征蒸馏模块执行下述操作：将所述多个不同尺度的精细特征图按照尺度从小到大进行融合，并加入注意力机制，得到融合特征图；将所述融合特征图通过卷积操作进行特征重构，得到多个不同尺度的新特征图；将所述多个不同尺度的新特征图进行融合，生成待提取特征图。