面向特定场景的行为识别算法的研究

2020-09-23王华锋

科学技术创新 2020年28期

关键词：关节点骨架权重

王华锋张鹏

（北方工业大学信息学院，北京100144）

1 方法

1.1 融合骨架信息的卷积神经网络

人体的骨架图是对人体轮廓的抽象，是一种包含了人体主要关节点信息和骨架连接结构信息的图形。与传统的卷积神经网络不同的是，本此研究提出的网络输入中包含了人体的骨架信息。其特征用如下公式来表示：

其中V 为骨架图中所有关节点的集合，包含了两个子集，第一个子集描述每个帧的骨架内连接，第二个子集包含帧间边缘信息，代表了关节随着时间的运动轨迹。在此网络的传播规则如下：

Wi表示第i 层的权重矩阵，σ 表示ReLU 激活函数，I 为输入邻接矩阵，它是某一关节点和其他关节点连接特征的数学表示。首先在邻接矩阵左侧乘以特征矩阵，完成了特征的融合操作，然后再在右侧乘以权重矩阵，实现了传统卷积神经网络加权操作。具体到x 层卷积其输出为：

其中，p 为采样函数，表示对输入信息的特征提取，列举了位置x 的近邻区域，K 为卷积核的维度，加权函数w提供了一个权重向量，来计算其与经过采样的输入向量的内积。在网络中将骨骼节点的近邻划分为固定的K 个子集，W 通过索引一个K维的张量来实现。则上述公式被重新定义如下：

1.2 注意力机制的设计

本次研究在上述提到的骨架网络结构里创新性地引入了注意力模块，模块中包括了两种注意力机制：时间注意力机制和空间注意力机制。在空间维度上不同关节点之间存在连通，具有很强的动态性，可以用注意力机制去自适应调节空间维度上节点之间的动态相关性：

其中为前一层的输入，V 是骨架关节点的集合，W 为需要进行学习的参数。在S 中代表了索引i 的点和索引j 的点的相关性。在进行卷积操作时将使用邻接矩阵A 与空间注意力矩阵S 一起对两点之间的动态权重值进行更新。

图1 时空注意力机制示意图

时空注意力模块能够使网络对不同时刻的信息赋予不同的权重。在时间维度上，不同的帧之间的行为特征也存在着相关性，这些相关性在不同点之间也是时刻发生着变化，因此，网络使用一个时间注意力机制去调整数据在各个时间的权重：

1.3 数据集的建立

本研究中构建了自己的行为识别骨架数据集。数据集中主要有五类日常的动作：跌倒，坐下，站立，喝水，太极，其中每个动作有30-40 个视频剪辑，共计170 个。其中一部分视频片段在实验室拍摄完成，一部分从网络的视频中提取。首先通过ffmpeg将视频的帧率转到30FPS，并将视频的分辨率进行统一调整为340*256，然后开始对视频中的人体骨架信息进行提取，每个信息元组中包含三组数据包括人体关节点的2D 坐标和置信度，最终将其保存为json 文件。

2 实验

本文从传统的卷积神经网络模型出发，引入了骨架网络模型，并对日常中常见的动作进行了分类和识别。实验完成了行为动作的分类任务，但当检测目标在快速运动下的采样会出现模糊，导致骨架信息丢失。

图2 行为检测结果图（1）

图3 行为检测结果图（2）

如图所示，骨架关节点的亮度半径代表了其权重的大小，不同关节点在不同动作中的权重有高有低，其中的空间注意力模块决定了不同关节点在行为构成中占的权重，时间注意力决定了不同帧在行为识别中占的比重。

具体而言在跌倒中的肩部关节点的权重较高，关节点亮度范围较大；喝水的动作中手部和胳膊的关节点权重更高，而下肢的权重则相对较低。

在行为识别任务上与四种不同的算法进行了对比，Two Stream Networks[1]以堆叠的光流矢量的形式对运动特征进行了建模，然后通过两个单独的网络进行训练。C3D[2]利用3x3x3 的三维卷积网络在大规模有监督的数据集上进行训练。LRCN 网络[3]输入图片后先使用传统CNN 来提取输入图片的特征，然后送入后续的LSTM网络进行处理。TSN 网络[4]对整个视频进行稀疏采样，而不是采用单帧分析的方法，然后网络对各个片段进行动作类别的初步预测。可以看出，本文提出的模型在识别率上更有优势。