[发明专利]一种基于对抗网络学习的视频的车辆检测方法和计数方法

权利要求书

1.一种基于对抗网络学习的视频的车辆检测方法，包括以下步骤：

S1、车辆视频图像的获取；收集若干幅未处理的车辆视频图像作为训练图像，将训练图像归一化处理，得到多幅归一化后的车辆视频图像，作为车辆视频图像训练样本；

S2、获得车辆视频图像检测模型；对条件约束下的视频图像判别网络D(x,l)、视频图像生成网络G(z,l)进行构造与对抗训练；另外获取车辆视频图像负样本，所述车辆视频图像负样本为若干幅归一化后的非车辆视频图像；将车辆视频图像负样本与车辆视频图像训练样本输入到对抗训练后的视频图像判别网络D(x,l)中，从对抗训练后的视频图像判别网络D(x,l)中提取车辆视频图像对抗特征，组成对抗特征向量，将对抗特征向量经过分类器训练，获得车辆视频图像检测模型；

S3、待检测车辆视频图像的获取；收集若干幅未处理的待检测车辆视频图像，并将未处理的待检测车辆视频图像的大小归一化，得到多幅归一化后的待检测车辆视频图像；

S4、车辆在车辆视频图像中具体位置的标记；将待检测车辆视频图像输入到所述的车辆视频图像检测模型，进行车辆的检测，定位并标记出车辆在视频图像中的具体位置。

2.根据权利要求1所述的一种基于对抗网络学习的视频的车辆检测方法，其特征在于，步骤S2包括以下步骤：

S21、构造带条件约束的视频图像判别网络D(x,l)和构造l～pl(l)表示条件约束分布N(α，δ^2)；l～pl(l)的公式为：其中x表示图像数据，l表示的是服从高斯分布的条件参数；α、δ^2为分布的参数，设定为车辆姿态或车辆运动状态；

S22、构造带条件约束的视频图像生成网络G(z,l)，z～pz(z)表示高斯噪声分布N(μ，σ^2)；z～pz(z)的公式为：其中z表示高斯噪声分布，l表示条件约束分布且设定为车辆姿态分布或车辆运动状态分布；μ、σ^2为分布的参数，分别为高斯分布的期望和方差；

S23、条件约束下视频图像判别网络D(x,l)和视频图像生成网络G(z,l)的对抗训练，得到对抗训练后的待条件的约束的视频图像判别网络D(x,l)；

S24、获取车辆视频图像负样本；收集若干幅非车辆视频图像作为未处理的车辆视频图像负样本，收集的未处理的车辆视频图像负样本的焦点集中在视频中车辆之外的图像区域，将所有未处理的车辆视频图像负样本的大小归一化处理，得到多幅归一化后的车辆视频图像负样本；

S25、提取车辆视频图像对抗特征；将车辆视频图像训练样本及车辆视频图像负样本作为输入，输入到训练后的带条件约束的视频图像判别网络D(x,l)，并将带条件约束的视频图像判别网络D(x,l)的深度卷积神经网络的其中一层作为车辆视频图像对抗特征输出；

S26、收集车辆视频图像对抗特征，组成对抗特征向量；

S27、将对抗特征向量经过分类器训练，得到车辆视频图像检测模型。