[发明专利]图像检测方法、装置、计算机可读存储介质及电子设备

权利要求书

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

获取待检测图像；

对所述待检测图像进行特征提取，得到所述待检测图像的特征图；

通过路径聚合网络PAN对所述特征图进行特征融合，得到第一特征融合图；

通过单阶段检测器SSH对所述第一特征融合图进行特征融合，得到第二特征融合图；

根据所述第二特征融合图确定所述待检测图像中的手部预测框位置、手部预测框置信度、手部关键点坐标及类别信息，作为图像检测结果。

2.根据权利要求1所述的方法，其特征在于，所述对待检测图像进行特征提取，得到所述待检测图像的特征图包括：

通过YOLO网络对所述待检测图像进行特征提取，得到所述待检测图像的特征图。

3.根据权利要求2所述的方法，其特征在于，所述YOLO网络为YOLOv5网络；所述YOLOv5网络包括Stemblock结构、用于下采样的输出层、手部关键点预测分支及ShuffleNet。

4.根据权利要求1所述的方法，其特征在于，所述通过路径聚合网络PAN对所述特征图进行特征融合，得到第一特征融合图包括：

对所述特征图进行多次自底向上的卷积操作；每一次所述卷积操作用于使得所述特征图的尺寸减半、通道翻倍；

将每一次所述卷积操作后得到的特征图进行横向连接；

对横向连接后得到的特征图进行自底向上的特征融合，得到多帧不同大小的所述第一特征融合图。

5.根据权利要求1所述的方法，其特征在于，所述SSH包括第一支路、第二支路及第三支路，所述第一支路包括1个3x3的卷积核，所述第二支路包括2个3x3的卷积核，第三支路包括3个3x3的卷积核；

所述通过单阶段检测器SSH对多帧所述第一特征融合图进行特征融合，得到第二特征融合图包括：

通过所述第一支路、所述第二支路及所述第三支路分别对多帧所述第一特征融合图进行卷积，得到多帧所述第二特征融合图。

6.根据权利要求1所述的方法，其特征在于，所述根据所述第二特征融合图确定所述待检测图像中的手部预测框位置、手部预测框置信度、手部关键点坐标及类别信息包括：

将每帧所述第二特征融合图中的每一个感受野，分别映射为第一预定通道数、第二预定通道数及第三预定通道数，得到第一通道结果、第二通道结果及第三通道结果；

根据所述第一通道结果确定所述类别信息，根据所述第二通道结果确定所述手部预测框位置和手部预测框置信度，根据所述第三通道结果确定所述手部关键点坐标。

7.根据权利要求6所述的方法，其特征在于，所述根据所述第二通道结果确定所述手部预测框位置和手部预测框置信度包括：

根据所述第二通道结果确定每帧所述第二特征融合图中的初始手部预测框及初始手部预测框置信度；

选择所述初始手部预测框置信度最大的所述初始手部预测框作为候选手部预测框；

分别计算所述候选手部预测框与每一个其他初始手部预测框的交并比值；

根据所述候选手部预测框的框中心位置、所述初始手部预测框的框中心位置及所述交并比值确定所述手部预测框位置。

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

获取模块，用于获取待检测图像；

提取模块，用于对所述待检测图像进行特征提取，得到所述待检测图像的特征图；

第一融合模块，用于通过路径聚合网络PAN对所述特征图进行特征融合，得到第一特征融合图；

第二融合模块，用于通过单阶段检测器SSH对所述第一特征融合图进行特征融合，得到第二特征融合图；

处理模块，用于根据所述第二特征融合图确定所述待检测图像中的手部预测框位置、手部预测框置信度、手部关键点坐标及类别信息，作为图像检测结果。

9.一种非临时性计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现权利要求1-7中任一项所述方法的步骤。

10.一种电子设备，其特征在于，包括：

存储器，其上存储有计算机程序；

处理器，用于执行所述存储器中的所述计算机程序，以实现权利要求1-7中任一项所述方法的步骤。