会员体验
专利管家(专利管理)
工作空间(专利管理)
风险监控(情报监控)
数据分析(专利分析)
侵权分析(诉讼无效)
联系我们
交流群
官方交流:
QQ群: 891211   
微信请扫码    >>>
现在联系顾问~
首页 / 专利库 / 空中交通管制员 / 基于监控视频的空中交通管制员不安全行为分类方法

基于监控视频的空中交通管制员不安全行为分类方法

申请号 CN202210966731.6 申请日 2022-08-12 公开(公告)号 CN115410035A 公开(公告)日 2022-11-29
申请人 武汉理工大学; 发明人 高曙; 揣明瑞; 罗帆;
摘要 本发明提出了基于监控视频的空中交通管制员不安全行为分类方法及控制方法,包括:S1获取数据集;S2数据集处理:将数据集输入至目标检测层,定位每一个管制员在图像中的位置坐标,将每一个管制员目标从原图像中裁剪出来;S3搭建双线性池化网络;S4设计双域混合注意力机制;S5采用交叉熵作为损失函数,对搭建的网络模型进行训练;S6进行指标评估,指标评估达到设定值则训练完成,将待检测视频图像输入至基于双域混合注意力机制的双线性池化网络模型得到分类结果。本发明通过引入双域混合注意力机制增加对不安全行为分类有重要影响的局部分类特征权重,削减次要信息特征权重,达到提高管制员不安全行为分类精度的目的。
权利要求

1.一种基于监控视频的空中交通管制员不安全行为分类方法,其特征在于:所述方法包括如下步骤:S1获取数据集:采集空中交通管制员工作监控视频,对视频图像进行预处理,形成数据集;

S2数据集处理:将数据集输入至目标检测层,目标检测层定位每一个管制员在图像中的位置坐标,同时将每一个管制员目标从原图像中裁剪出来,作为后续分类模型的输入;

S3分类模型搭建:搭建双线性池化网络,使用主干网络EfficentNetV2对数据集中的图像进行分类处理,输出特征图F;

S4将管制员不安全行为分类作为细粒度图像分类,在双线性池化网络的基础上,针对图像局部分类特征不明显的特点,设计双域混合注意力机制:首先将所述特征图F输入至通道域注意力机制得到通道域注意力机制加权特征图Fc,再将所述通道域注意力机制加权特征图Fc输入至空间域注意力机制得到空间域注意力机制加权特征图Fs;

S5采用交叉熵作为损失函数,使用Pytorch深度学习框架对搭建的网络模型进行训练;

S6针对数据集,基于双域混合注意力机制的双线性池化网络生成预测pth文件,进行指标评估,指标评估达到设定值则训练完成,将待检测视频图像输入至基于双域混合注意力机制的双线性池化网络模型得到分类结果。

2.根据权利要求1所述的基于监控视频的空中交通管制员不安全行为分类方法,其特征在于:步骤S4中通道域注意力机制的处理过程为:

401)使用全局最大池化和全局平均池化并行处理输入特征图F,得到最大池化特征FGMPC×1×1 C×1×1∈R 和平均池化特征FGAP∈R :

FGMP=GMP(F),FGAP=GAP(F)    (4)其中GMP()表示全局最大池化,GAP()表示全局平均池化;

402)将全局最大池化、全局平均池化特征图输入共享权重的多层感知机MLP进行缩放C×1×1 C×1×1映射学习后,得到通道域特征C1∈R 、C2∈R :

C1=f′(f(FGMP)),C2=f′(f(FGAP))    (5)其中,f′(f(·))表示全连接层的缩放映射学习;

403)对全局最大池化、全局平均池化特征并行处理,其中一路通过特征元素拼接,将C1和C2进行融合,再经过全连接层做线性映射处理,另一路通过特征元素相加,将C1和C2进行融合,最后,再将两者以特征元素相加的方式融合并进行Sigmod归一化,此时得到通道域注C×1×1意力权重向量Mc∈R :

其中,Sigmoid表示激活函数, 表示特征元素相加,concat表示特征拼接,FC表示全连接操作;

404)将通道域注意力权重向量Mc和输入特征图F进行矩阵点乘运算,最终得到与原始输入特征图F尺寸相同的通道域注意力加权特征图Fc其中, 表示矩阵点乘运算。

3.根据权利要求1所述的基于监控视频的空中交通管制员不安全行为分类方法,其特征在于:步骤S4中空间域注意力机制的处理过程为:

405)采用与通道域相同的全局最大池化和全局平均池化并行处理输入的通道域加权C×W×H C×1×1 C×1×1特征图Fc∈R ,生成最大池化特征FG′MP∈R 和平均池化特征FG′AP∈R :FG′MP=GMP(Fc),FG′AP=GAP(Fc)    (8)

406)并行处理最大池化特征FG′MP和平均池化特征FG′AP,其中一路沿通道方向将FG′MP和

2×H×W

FG′AP拼接融合,得到特征图M1∈R ,另一路通过特征元素相加将两个池化特征进行融

2×H×W 2×H×W

合,得到特征图M2∈R ,将特征图M1和M2沿通道方向拼接融合,到特征图M3∈R ,将特

1×H×W

征图M3经过下采样得到空间域特征M∈R ,最后通过Sigmod归一化处理,得到空间域注

1×H×W

意力权重矩阵Ms∈R :

407)将空间域注意力权重矩阵Ms和输入的通道域加权特征图Fc进行矩阵点乘运算,得到空间域注意力加权特征图Fs:

4.根据权利要求1所述的基于监控视频的空中交通管制员不安全行为分类方法,其特征在于:步骤S5中首先利用注意力模块特征提取器提取预测向量Fs,然后利用初始图片的y y标签y训练特征,则相应的交叉熵损失函数为‑log(Fs),其中Fs代表图像在实际类别y上的预测概率,则在训练特征时损失函数为:y

H=‑log(Fs)    (11)

5.根据权利要求3所述的基于监控视频的空中交通管制员不安全行为分类方法,其特征在于:步骤S3中对数据集进行分类处理的过程为:将数据集图像经过两个相同的卷积神h×w×c h×w×c经网络输出的卷积层分别为X1∈R 、X2∈R ,其中,h、w、c分别代表特征图的高、宽、通道数,Xi(i=1,2)位于空间位置p处的描述算子分别为 和通过在每个空间位置上使用矩阵外积组合得到双线性特征,则双线性池化在位置p处的特征表示为:

其中,Fp是双线性模型在位置p处的输出结果,则双线性模型表示为:由于采用两个相同的卷积神经网络,所以双线性池化模型表示为:对于一个卷积神经网络有c个通道数,则位置p处的特征维度为1×c,然后与另一个卷积神经网络在相同位置处得到的1×c的矩阵进行乘积,得到的矩阵维度为c×c,之后将每个位置处得到的矩阵求和,转换为向量形式,得到图像上每个位置的双线性特征。

6.根据权利要求5所述的基于监控视频的空中交通管制员不安全行为分类方法,其特征在于:步骤S1中使用Opencv将获取到的数据集中的图像尺寸调整为512×512,将数据集中的一些过度曝光图像进行去噪处理,步骤S2中所述目标检测层利用YOLOv5网络检测并定位视频图像中的所有管制员前景区域,将其从原图像中分割出来,作为后续模型的输入。

7.根据权利要求6所述的基于监控视频的空中交通管制员不安全行为分类方法,其特征在于:步骤S5中使用公共数据集COCO2017对YOLOv5网络训练,训练过程中参数设置为:选择随机梯度下降作为优化器,权重衰减weight_decay=0.0005,学习率learning_rate=

0.001,动量momentum=0.9,批量尺寸batch_size=16,迭代次数epochs=300。

8.根据权利要求6所述的基于监控视频的空中交通管制员不安全行为分类方法,其特征在于:步骤S5中训练过程包括微调训练阶段,将主干网络部分的head权重冻结,并将双线性池化网络全连接层输出维度改为所需分类的类别数量,包含正常行为和六类脱岗行为;

使用管制员不安全行为数据集进行微调训练,微调训练阶段参数设置为:选择随机梯度下降作为优化器,权重衰减weight_decay=0.0005,学习率learning_rate=0.001,动量momentum=0.9,批量尺寸batch_size=16,迭代次数epochs=100。

9.根据权利要求5所述的基于监控视频的空中交通管制员不安全行为分类方法,其特征在于:步骤S6中采用精准率P作为评价指标,计算方法为:其中,TP、FP分别表示在某类不安全行为的分类结果里,正确和错误分为该类不安全行为的数量,评价指标精准率以百分制表示。

10.一种计算机设备,其包括存储器、处理器和存储在存储器中可供处理器运行的程序指令,其中所述处理器执行所述程序指令以实现权利要求1至9中任一项所述的方法。

说明书全文

基于监控视频的空中交通管制员不安全行为分类方法

技术领域

[0001] 本发明属于计算机视觉领域,具体是指一种基于目标检测和双域混合注意力池化网络的基于监控视频的空中交通管制员不安全行为分类方法。

背景技术

[0002] 随着科学技术的发展,航空业以其安全,快捷的特点在交通领域发挥越来越重要的作用。其中,空中交通管制员(以下简称为“管制员”)是航空运输体系的重要参与人员,主要负责维护空中交通运行秩序,保障航空运输体系的运行安全。鉴于管制员在航空运输体系中的重要地位,要求其在工作期间保持精神高度集中,并且管制员的工作制度为昼夜轮班,导致管制员易处于疲劳状态。为了保障飞行安全,遏制不安全事件的发生,鉴于管制员工作的重要性、特殊性,需要密切关注其行为的规范和要求,严格防止出现与规范要求不相符的不安全行为。目前,对管制员的不安全行为检测主要利用监控视频,采用人工巡查方式进行识别。人工巡查需要对监控录像进行回放、人工识别和审核,增加了额外的劳动成本,使得管制员视频监控系统没有得到充分利用和发挥有效作用。因此,基于视频监控数据,研究可以高效、精准地自动识别管制员不安全行为的方法,将有助于对管制员不安全行为预警并采取防范措施,排除潜在隐患,保障空中交通安全,具有重要的现实意义与应用价值。
[0003] 目前利用视频数据分析管制员不安全行为鲜有报道,而在煤矿、建筑工程等领域,对不安全行为分类的研究主要基于大量不安全行为样本数据进行训练,与不安全行为偶发、稀疏等特点不符,因此,传统的图像分类模型在管制员工作场景下会出现较高的漏检与误检,难以进行实际应用。并且管制员工作场景与煤矿、建筑行业有显著不同,一般存在多名管制员,此时视频图像中可能出现多种行为类别,导致图像中存在大量的信息冗余。如果不首先进行管制员检测与分割,这些冗余信息带来的噪声干扰将不利于模型对不同行为特征的理解,容易产生错误的推理结果,降低分类准确率。此外,管制员不安全行为之间相似度较高,且不安全行为与某些正常行为之间区分度不大,即不安全行为的局部分类特征不明显,这些也都影响现有模型的分类效果。

发明内容

[0004] 本发明的目的在于克服现有技术应用于管制员工作场景的不足,而提出一种基于目标检测和双域混合注意力池化网络的基于监控视频的空中交通管制员不安全行为分类方法,解决现有不安全行为分类方法在管制员工作场景下分类准确率较低的问题。
[0005] 为实现上述目的,本发明所设计的基于监控视频的空中交通管制员不安全行为分类方法,其特殊之处在于,所述方法包括如下步骤:
[0006] S1获取数据集:采集空中交通管制员工作监控视频,对视频图像进行预处理,形成数据集;
[0007] S2数据集处理:将数据集输入至目标检测层,目标检测层定位每一个管制员在图像中的位置坐标,同时将每一个管制员目标从原图像中裁剪出来,作为后续分类模型的输入;
[0008] S3分类模型搭建:搭建双线性池化网络,使用主干网络EfficentNetV2对数据集中的图像进行分类处理,输出特征图F;
[0009] S4将管制员不安全行为分类作为细粒度图像分类,在双线性池化网络的基础上,针对图像局部分类特征不明显的特点,设计双域混合注意力机制:首先将所述特征图F输入至通道域注意力机制得到通道域注意力机制加权特征图Fc,再将所述通道域注意力机制加权特征图Fc输入至空间域注意力机制得到空间域注意力机制加权特征图Fs;
[0010] S5采用交叉熵作为损失函数,使用Pytorch深度学习框架对搭建的网络模型进行训练;
[0011] S6针对数据集,基于双域混合注意力机制的双线性池化网络生成预测pth文件,进行指标评估,指标评估达到设定值则训练完成,将待检测视频图像输入至基于双域混合注意力机制的双线性池化网络模型得到分类结果。
[0012] 优选地,步骤S4中通道域注意力机制的处理过程为:
[0013] 401)使用全局最大池化和全局平均池化并行处理输入特征图F,得到最大池化特C×1×1 C×1×1征FGMP∈R 和平均池化特征FGAP∈R :
[0014] FGMP=GMP(F),FGAP=GAP(F)   (4)
[0015] 其中GMP()表示全局最大池化,GAP()表示全局平均池化,C、R分别代表特征图的通道数和实数集;
[0016] 402)将全局最大池化、全局平均池化特征图输入共享权重的多层感知机MLP进行C×1×1 C×1×1缩放映射学习后,得到通道域特征C1∈R 、C2∈R :
[0017] C1=f′(f(FGMP)),C2=f′(f(FGAP))   (5)
[0018] 其中,f′(f(·))表示全连接层的缩放映射学习;
[0019] 403)对全局最大池化、全局平均池化特征并行处理,其中,一路通过特征元素拼接,将C1和C2进行融合,再经过全连接层做线性映射处理,另一路通过特征元素相加,将C1和C2进行融合,最后,再将两者以特征元素相加的方式融合并进行Sigmod归一化,此时得到通C×1×1道域注意力权重向量Mc∈R :
[0020]
[0021] 其中,Sigmoid表示激活函数, 表示特征元素相加,concat表示特征拼接,FC表示全连接操作;
[0022] 404)将通道域注意力权重向量Mc和输入特征图F进行矩阵点乘运算,最终得到与原始输入特征图F尺寸相同的通道域注意力机制加权特征图Fc:
[0023]
[0024] 其中, 表示矩阵点乘运算。
[0025] 优选地,步骤S4中空间域注意力机制的处理过程为:
[0026] 405)采用与通道域相同的全局最大池化和全局平均池化并行处理输入的通道域C×W×H C×1×1注意力机制加权特征图Fc∈R ,生成最大池化特征F′GMP∈R 和平均池化特征F′GAP∈C×1×1
R :
[0027] F′GMP=GMP(Fc),F′GAP=GAP(Fc)   (8)
[0028] 406)并行处理最大池化特征F′GMP和平均池化特征F′GAP,其中一路沿通道方向将2×H×W
F′GMP和F′GAP拼接融合,得到特征图M1∈R ,另一路通过特征元素相加将两个池化特征进
2×H×W 2×H×W
行融合,得到特征图M2∈R ,将特征图M1和M2沿通道方向拼接融合,到特征图M3∈R ,
1×H×W
将特征图M3经过下采样得到空间域特征M∈R ,最后通过Sigmod归一化处理,得到空间
1×H×W
域注意力权重矩阵Ms∈R :
[0029] Ms=Sigmoid(f(Concat(Concat(F′GMP,F′GAP),(F′GMP⊕F′GAP))))   (9)[0030] 407)将空间域注意力权重矩阵Ms和输入的通道域注意力机制加权特征图Fc进行矩阵点乘运算,得到空间域注意力加权特征图Fs:
[0031]
[0032] 优选地,步骤S5中首先利用注意力模块特征提取器提取预测向量Fs,然后利用初y y始图片的标签y训练特征,则相应的交叉熵损失函数为‑log(Fs),其中Fs代表图像在实际类别y上的预测概率,则在训练特征时损失函数为:
[0033] H=‑log(Fsy)   (11)
[0034] 优选地,步骤S3中对数据集进行分类处理的过程为:将数据集图像输入两个的卷h×w×c h×w×c积神经网络,卷积层的输出分别为X1∈R 、X2∈R ,其中,h、w、c分别代表特征图的高、宽、通道数,Xi(i=1,2)位于空间位置p处的描述算子分别为 和
通过在每个空间位置上使用矩阵外积组合得到双线性特征,则双线性
池化在位置p处的特征表示为:
[0035]
[0036] 其中,Fp是双线性模型在位置p处的输出结果,则双线性模型表示为:
[0037]
[0038] 由于采用两个相同的卷积神经网络,所以双线性池化模型表示为:
[0039]
[0040] 对于一个卷积神经网络有c个通道数,则位置p处的特征维度为1×c,然后与另一个卷积神经网络在相同位置处得到的1×c的矩阵进行乘积,得到的矩阵维度为c×c,之后将每个位置处得到的矩阵求和,转换为向量形式,得到图像上每个位置的双线性特征。
[0041] 优选地,步骤S1中使用Opencv将获取到的数据集中的图像尺寸调整为512×512,将数据集中的一些过度曝光图像进行去噪处理,步骤S2中所述目标检测层利用YOLOv5网络检测并定位视频图像中的所有管制员前景区域,将其从原图像中分割出来,作为后续模型的输入。
[0042] 优选地,步骤S2中使用公共数据集COCO2017对YOLOv5网络训练,训练过程中参数设置为:选择随机梯度下降作为优化器,权重衰减weight_decay=0.0005,学习率learning_rate=0.001,动量momentum=0.9,批量尺寸batch_size=16,迭代次数epochs=300。
[0043] 优选地,步骤S2中训练过程包括微调训练阶段,将主干网络部分的head权重冻结,并将双线性池化网络全连接层输出维度改为所需分类的类别数量,包含正常行为和六类不安全行为;使用管制员不安全行为数据集进行微调训练,微调训练阶段参数设置为:选择随机梯度下降作为优化器,权重衰减weight_decay=0.0005,学习率learning_rate=0.001,动量momentum=0.9,批量尺寸batch_size=16,迭代次数epochs=100。
[0044] 优选地,步骤S6中采用精准率P作为评价指标,计算方法为:
[0045]
[0046] 其中,TP、FP分别表示在某类不安全行为的分类结果里,正确和错误分为该类不安全行为的数量,评价指标精准率百分制表示。
[0047] 本发明还提出一种计算机设备,其包括存储器、处理器和存储在存储器中可供处理器运行的程序指令,其中所述处理器执行所述程序指令以实现上述一种基于监控视频的空中交通管制员不安全行为分类方法。
[0048] 本发明是通过如下的技术方案实现的:
[0049] 1)基于目标检测和双域混合注意力池化网络的管制员不安全行为分类方法,在原有的双线性池化网络基础上,添加了目标检测网络和双域混合注意力机制。
[0050] 2)针对管制员工作场景中存在多名管制员的特点,首先,使用YOLOv5网络定位管制员在图像中的位置坐标,同时将管制员目标从原图像中裁剪出来,作为后续分类模型的输入,从而将多目标分类任务转变为单目标分类任务,有效滤除冗余信息,同时降低图像中的噪声信息。
[0051] 3)针对图像局部分类特征不明显,即不安全行为之间相似度高的问题,本发明将管制员不安全行为分类看作细粒度图像分类问题,在双线性池化网络的基础上,针对图像局部分类特征不明显的特点,设计通道域和空间域的混合注意力机制。
[0052] 相比于现有技术,本发明具有如下优点:
[0053] 1)利用YOLOv5网络对视频图像中管制员进行有效分割。管制员工作场景的特点之一是场景中存在多名管制员,因此同一图像中可能出现多种行为类别,如果不首先进行分割,将不利于网络分别对不同行为特征的理解,从而产生错误的推理结果。相较于传统方法,该方法引入YOLOv5目标检测网络对原始图像进行目标检测和区域分割,将原始图像划分为与管制员人数相等的单人图像,降低冗余信息和噪声给分类任务造成的影响,加强分类模型对管制员行为特征的理解,为不安全行为分类效果的提高奠定了基础。
[0054] 2)利用双域混合注意力机制增强局部分类特征。通过引入双域混合注意力机制可以增加对不安全行为分类有重要影响的局部分类特征权重,削减次要信息特征权重,即引导模型关注动作幅度小、行为相似度高的不安全行为之间的细微差异,从而达到提高管制员不安全行为分类精度的目的。

附图说明

[0055] 图1为基于目标检测和双域混合注意力池化的不安全行为模型架构图。
[0056] 图2为基于YOLOv5的管制员目标检测层结构
[0057] 图3为管制员工作场景图。
[0058] 图4为管制员不安全行为与正常行为示意图。
[0059] 图5为引入YOLOv5前后分类效果可视化图。
[0060] 图6为融合双域混合注意力机制的双线性池化结构。
[0061] 图7为通道域注意力机制结构示意图。
[0062] 图8为空间域注意力机制结构示意图。
[0063] 图9为看书不安全行为CAM图。
[0064] 图10为摘耳机不安全行为CAM图。
[0065] 图11为玩手机不安全行为CAM图。
[0066] 图12为睡岗不安全行为CAM图。
[0067] 图13为聊天不安全行为CAM图。
[0068] 图14为各类不安全行为分布图。

具体实施方式

[0069] 为使本发明要解决的技术问题、技术方案和优点更加清楚,下面将结合附图及具体实施例进行详细描述,但以下实施例仅是说明性的,本发明的保护范围并不受这些实施例的限制。
[0070] 基于目标检测和双域混合注意力池化的不安全行为模型架构如图1所示,本发明提出的一种基于监控视频的空中交通管制员不安全行为分类方法,包括如下步骤:
[0071] S1获取数据集:采集空中交通管制员工作监控视频,对视频图像进行预处理,形成数据集。
[0072] 鉴于管制员工作的特殊性与保密性,所构建的实验数据集均采自某空管分局管制员培训室(与实际工作场景一致)的监控视频。视频帧率为12fps,分辨率为720×480,剪辑掉无用片段后,得到视频总时长约500分钟,其中不安全行为占43分钟,其余皆为正常工作行为。按照5帧间隔采集图像,并将其按照8:2的比例划分为训练集和测试集。
[0073] S2数据集处理::将数据集输入至目标检测层,目标检测层定位每一个管制员在图像中的位置坐标,同时将每一个管制员目标从原图像中裁剪出来,作为后续分类模型的输入。
[0074] 为方便模型训练,同时降低网络计算复杂度,使用Opencv将S1中获取到的图像尺寸调整为512×512,将数据集中的一些过度曝光图像进行去噪处理,利用如图2所示的YOLOv5网络检测并定位视频图像中的所有管制员前景区域,将其从原图像中分割出来,作为后续模型的输入。分割前后的数据集示意图详见图3、图4。
[0075] S3分类模型搭建:搭建双线性池化网络,使用主干网络EfficentNetV2对数据集中的图像进行分类处理,输出特征图F。
[0076] 对数据集进行分类处理的过程为:将数据集图像输入两个的卷积神经网络,卷积h×w×c h×w×c层的输出分别为X1∈R 、X2∈R ,其中,h、w、c分别代表特征图的高、宽、通道数,Xi(i=1,2)位于空间位置p处的描述算子分别为 和 通过在
每个空间位置上使用矩阵外积组合得到双线性特征,则双线性池化在位置p处的特征表示为:
[0077]
[0078] 其中,Fp是双线性模型在位置p处的输出结果,则双线性模型表示为:
[0079]
[0080] 由于采用两个相同的卷积神经网络,所以双线性池化模型表示为:
[0081]
[0082] 对于一个卷积神经网络有c个通道数,则位置p处的特征维度为1×c,然后与另一个卷积神经网络在相同位置处得到的1×c的矩阵进行乘积,得到的矩阵维度为c×c,之后将每个位置处得到的矩阵求和,转换为向量形式,得到图像上每个位置的双线性特征。
[0083] 为了验证引入目标检测层YOLOv5网络对管制员不安全行为分类的有效性,设计了对比实验:首先,使用管制员不安全行为分类数据集的原始图像训练图像分类模型,得到实验评价指标。然后,将YOLOv5网络引入上述图像分类模型,并使用管制员单人数据集的单人图像重新训练。最后,均在管制员不安全行为分类数据集上进行测试,通过对比上述两种模型在管制员不安全行为分类数据集上的实验评价指标,验证目标检测方法能有效提高管制员不安全行为分类准确率。实验对比引入YOLOv5网络前后的分类模型对管制员不安全行为的分类效果。以看书和玩手机两种不安全行为为例,在不安全行为分类数据集中选取部分图像测试,可视化结果如图5所示。图5中,(a)、(c)为EfficentNetv2模型直接分类的可视化结果,(b)、(d)为引入YOLOv5网络后EfficentNetv2模型分类的可视化结果,每个图片左下角标出了图片中各管制员的不安全行为分类结果,浅色字体表示分类错误,深色字体表示分类正确,正常工作且分类为正常行为的管制员未标出。从分类结果来看,(a)中部分看书行为被错误分类为玩手机和正常行为,(c)为玩手机不安全行为,可以看出手机目标较小,并且被手掌区域遮盖,部分被错误分类为正常行为;(b)、(d)利用YOLOv5将管制员分割为单人图像后进行分类,有效缓解了误识别问题,并且可以同时识别多种不安全行为。本发明引入YOLOv5的目标检测网络,在管制员不安全行为数据集上的指标评价结果如表1所示,评价指标精准率以百分制显示。由于本发明利用YOLOv5网络检测管制员人数,进而判断是否发生脱岗不安全行为,故未引入YOLOv5网络的分类方法计算脱岗行为的分类准确率,因此表1中相应地方用“‑”表示。
[0084] 表1基于YOLOv5的目标检测网络评估实验
[0085]
[0086] 综合图5和表1的实验数据可知,使用YOLOv5网络首先对原始图像进行处理,将分割后的管制员单人图像作为分类网络的输入,对管制员不安全行为分类效果有明显提升。其原因分析如下:
[0087] 管制员工作场景的特点之一是同一图像中存在多名管制员,若直接对不安全行为分类数据集的原始图像进行训练,同一图像中可能出现多种行为类别,给分类任务带来了大量信息冗余和噪声干扰,不利于网络的特征理解,从而产生错误的推理结果。将YOLOv5网络引入分类模型,对原始图像进行目标检测和区域裁剪,将管制员从原始图像中分割出来,再将其作为分类模型的输入,更有利于模型对管制员各种行为特征的理解,同时也能精确定位到做出不安全行为的管制员。以上实验验证了将YOLOv5目标检测网络引入不安全行为分类模型,是符合管制员工作场景特点的有效改进措施。
[0088] S4将管制员不安全行为分类作为细粒度图像分类,在双线性池化网络的基础上,针对图像局部分类特征不明显的特点,设计双域混合注意力机制:第一条通道针对通道特征,第二条通道针对空间特征,融合双域混合注意力机制的双线性池化结构如图6所示。具体步骤包括:设计通道域注意力机制,如图7所示,处理过程为:
[0089] 401)使用全局最大池化和全局平均池化并行处理输入特征图F,得到最大池化特C×1×1 C×1×1征FGMP∈R 和平均池化特征FGAP∈R :
[0090] FGMP=GMP(F),FGAP=GAP(F)   (4)
[0091] 其中GMP()表示全局最大池化,GAP()表示全局平均池化;
[0092] 402)将全局最大池化、全局平均池化特征图输入共享权重的多层感知机MLP进行C×1×1 C×1×1缩放映射学习后,得到通道域特征C1∈R 、C2∈R :
[0093] C1=f′(f(FGMP)),C2=f′(f(FGAP))   (5)
[0094] 其中,f′(f(·))表示全连接层的缩放映射学习;
[0095] 403)对全局最大池化、全局平均池化特征并行处理,其中一路通过特征元素拼接,将C1和C2进行融合,再经过全连接层做线性映射处理,另一路通过特征元素相加,将C1和C2进行融合,最后,再将两者以特征元素相加的方式融合并进行Sigmod归一化,此时得到通道C×1×1域注意力权重向量Mc∈R :
[0096]
[0097] 其中,Sigmoid表示激活函数, 表示特征元素相加,concat表示特征拼接,FC表示全连接操作,C1、C2表示通道域特征;
[0098] 404)将通道域注意力权重向量Mc和输入特征图F进行矩阵点乘运算,最终得到与原始输入特征图F尺寸相同的通道域注意力加权特征图Fc
[0099]
[0100] 其中, 表示矩阵点乘运算。
[0101] 通道域注意力机制通过关注特征图通道域维度的不同特征分布情况,计算相应的特征权重,以提高特征图的表征能力。为了进一步捕获空间位置之间的依赖关系,设计了空间域注意力机制,如图8所示:
[0102] 405)采用与通道域相同的全局最大池化和全局平均池化并行处理输入的通道域C×W×H C×1×1 C×1×1加权特征图Fc∈R ,生成最大池化特征F′GMP∈R 和平均池化特征F′GAP∈R :
[0103] F′GMP=GMP(Fc),F′GAP=GAP(Fc)   (8)
[0104] 其中,GMP表示全局最大池化,GAP表示全局平均池化。
[0105] 406)并行处理最大池化特征F′GMP和平均池化特征F′GAP,其中一路沿通道方向将2×H×W
F′GMP和F′GAP拼接融合,得到特征图M1∈R ,另一路通过特征元素相加将两个池化特征进
2×H×W 2×H×W
行融合,得到特征图M2∈R ,将特征图M1和M2沿通道方向拼接融合,到特征图M3∈R ,
1×H×W
将特征图M3经过下采样得到空间域特征M∈R ,最后通过Sigmod归一化处理,得到空间
1×H×W
域注意力权重矩阵Ms∈R :
[0106] Ms=Sigmoid(f(Concat(Concat(F′GMP,F′GAP),(F′GMP⊕F′GAP))))   (9)[0107] 其中,Sigmoid表示激活函数, 表示特征元素相加,concat表示特征拼接。
[0108] 407)将空间域注意力权重矩阵Ms和输入的通道域加权特征图Fc进行矩阵点乘运算,得到空间域注意力加权特征图Fs:
[0109]
[0110] 得到的注意力特征图Fs即为输出特征图。
[0111] 为了验证融合双域混合注意力机制对管制员不安全行为分类的有效性,设计了对比实验:模型其他结构保持不变,对比引入双域混合注意力机制前后的实验效果。通过对比在管制员不安全行为分类数据集上的评价指标,验证引入双域混合注意力机制对管制员不安全行为分类的有效性。为了验证引入双域混合注意力机制对图像特征描述的影响,图9至图13展示了引入注意力机制前后的五种不安全行为类激活图(Class Activation Mapping,以下简称“CAM图”),CAM图可以反应分类网络所关注的图像区域,即图像中的深色区域。从图9至图13可以看出,未引入双域混合注意力机制前,模型对图像的关注区域较为分散。而引入双域混合注意力机制后,模型则更关注与当前分类行为相关的特征,如图9对看书不安全行为的关注区域更集中于手部,图12对睡岗不安全行为的关注更集中于头部和背部,图13对聊天不安全行为的关注更集中于嘴部。由此可知,引入双域混合注意力机制能较好的提升管制员不安全行为分类效果。本发明提出的双域混合注意力机制在管制员不安全行为数据集上的指标评价结果如表2所示,评价指标精准率以百分制显示。
[0112] 表2融合双域混合注意力机制评估实验
[0113]
[0114] 结合图9至图13、表2实验结果分析可以看出,相比于未引入注意力机制,融合双域混合注意力机制的双线性池化层表现更好,可能原因分析如下:
[0115] (1)引入双域混合注意力机制之后的双线性池化网络信息处理能力更强,能更加关注重要特征(动作特征)而抑制不重要特征,进而加强了对动作幅度小、行为相似度高的管制员不安全行为分类效果。
[0116] (2)双域混合注意力机制同时结合了通道域注意力和空间域注意力,相较于单域注意力机制,双域混合注意力机制能进一步提升映射特征的表征能力,通过捕获空间位置之间以及通道图之间的依赖关系,挖掘双域重要特征信息,从而生成更具有辨识度的特征表示,提升不安全行为分类性能。
[0117] S5采用交叉熵作为损失函数,使用Pytorch深度学习框架对搭建的网络模型进行训练。
[0118] 交叉熵反映了实际输出与期望输出的接近程度,首先利用注意力模块特征提取器提取预测向量Fs,然后利用初始图片的标签y来训练特征,则相应的交叉熵损失函数为‑logy y(Fs),其中Fs代表图像在实际类别y上的预测概率。在训练特征时损失函数为:
[0119] H=‑log(Fsy)   (11)
[0120] 使用Pytorch深度学习框架对搭建的网络模型进行训练,训练包括如下步骤:
[0121] (1)YOLOv5网络训练:对自制数据集标注检测框非常耗时,但考虑到人体结构的高度相似性,在公共数据集上训练的目标检测网络理论上也可以适用于管制员。使用公共数据集COCO2017对YOLOv5网络训练,训练过程中参数设置如下:选择随机梯度下降作为优化器,权重衰减weight_decay=0.0005,学习率learning_rate=0.001,动量momentum=0.9,批量尺寸batch_size=16,迭代次数epochs=300。
[0122] (2)融合双域混合注意力机制的双线性池化网络训练:通过对比网络性能,选用EfficentNetv2网络作为融合双域混合注意力机制的双线性池化网络(以下简称“双线性池化网络”)的主干网络。训练过程分为预训练和微调训练两个阶段。
[0123] 预训练阶段,使用ImageNet2012数据集训练双线性池化网络,预训练过程中参数设置如下:选择随机梯度下降作为优化器,权重衰减weight_decay=0.0001,学习率learning_rate=0.001,动量momentum=0.9,批量尺寸batch_size=16,迭代次数epochs=300。
[0124] 微调训练阶段,遵循迁移学习思路,将主干网络部分的head权重冻结,并将双线性池化网络全连接层输出维度改为所需分类的类别数量。使用管制员不安全行为数据集进行微调训练,微调训练阶段参数设置除迭代次数epochs=100,其他与预训练阶段一致。
[0125] S6针对数据集,基于双域混合注意力机制的双线性池化网络生成预测pth文件,进行指标评估,指标评估达到设定值则训练完成,将待检测视频图像输入至基于双域混合注意力机制的双线性池化网络模型得到分类结果。
[0126] 本发明使用的训练数据均采自某空管分局管制员培训室。数据集共有图像5668张,其中脱岗792张、睡岗572张、使用非工作所需电子产品(玩手机)990张、看书475张、未佩戴耳机(或摘耳机)728张、聊天1086张,此外需要分类模型也能对正常行为进行分类,所以采集了部分正常行为图像1025张,管制员行为类型分布图如图14所示。
[0127] 为了评估模型性能,采用精准率P(Precision)作为评价指标。精准率又称为查准率,表示正确预测为正例的样本占所有被预测为正例的样本的比例,度量分类器的准确性,计算方法为:
[0128]
[0129] 其中,TP、FP分别表示在某类不安全行为的分类结果里,正确和错误分为该类不安全行为的数量。
[0130] 本发明的指标评估实验结果如表3所示,评价指标精准率以百分制显示。
[0131] 表3管制员不安全行为分类模型指标评估实验结果
[0132]
[0133] 本发明还提出一种计算机设备,其包括存储器、处理器和存储在存储器中可供处理器运行的程序指令,其中所述处理器执行所述程序指令以实现上述的方法。
[0134] 以上应用了具体个例对本发明进行阐述,只是用于帮助理解本发明,并不用以限制本发明。对于本发明所属技术领域的技术人员,依据本发明的思想,还可以做出若干简单推演、变形或替换。本领域的技术人员容易理解,以上所述仅为本发明专利的较佳实施例而已,并不用以限制本发明专利,凡在本发明专利的精神和原则之内所作的任何修改、等同替换和改进等,均应包含在本发明专利的保护范围之内。
[0135] 本说明书未作详细描述的内容属于本领域专业技术人员公知的现有技术。