[发明专利]基于生成对抗网络的人群监控方法、装置、设备及介质

权利要求书

1.一种基于生成对抗网络的人群监控方法，包括：

训练生成对抗网络，所述生成对抗网络包括生成器模型和判别器模型，其中，在训练过程中利用至少一个第一人群图像对所述生成对抗网络反复执行如下操作，直到所述判别器模型判别出的相似性满足预设阈值条件，包括：

基于对所述第一人群图像中的人群标定和变换处理，获得所述第一人群密度真值图；

以第一人群图像作为所述生成器模型的输入，训练所述生成器模型输出第一人群密度估计图；

在训练过程中以所述第一人群密度真值图和所述第一人群密度估计图作为所述判别器模型的输入，以使所述判别器模型判别所述第一人群密度真值图和所述第一人群密度估计图的相似性；

在训练结束后，以目标监控区域的第二人群图像作为所述生成器模型的输入，基于所述生成器模型输出的第二人群密度估计图来估计所述目标监控区域中的人数。

2.根据权利要求1所述的方法，其中，所述以第一人群图像作为所述生成器模型的输入，训练所述生成器模型输出第一人群密度估计图还包括：

以所述第一人群图像和所述第一人群密度真值图拼接后的图像，作为所述生成器模型的输入。

3.根据权利要求1所述的方法，其中，所述基于对所述第一人群图像中的人群标定和变换处理，获得所述第一人群密度真值图包括：

对所述第一人群图像以人头为中心进行标定，得到标定图像；以及

将标定图像进行高斯平滑处理，得到所述第一人群密度真值图。

4.根据权利要求1所述的方法，其中，所述训练生成对抗网络还包括获取至少一个所述第一人群图像，具体包括：

从人群监控视频中截取图像帧；

从所述图像帧中裁剪尺寸为预设大小的至少一个图像块；以及

基于所述至少一个图像块，得到至少一个所述第一人群图像。

5.根据权利要求1所述的方法，其中，所述方法还包括：

基于在人群监控视频中划定所述目标监控区域的操作，从所述人群监控视频中截取所述第二人群图像。

6.根据权利要求5所述的方法，其中，所述从所述人群监控视频中截取所述第二人群图像还包括：

每间隔固定帧数，从所述人群监控视频中自动截取所述第二人群图像。

7.根据权利要求1或6所述的方法，其中，所述方法还包括：

当估计出所述目标监控区域中的人数大于或等于预设的告警人数时，实时告警。

8.一种基于生成对抗网络的人群监控装置，包括：

训练模块，用于训练生成对抗网络，所述生成对抗网络包括生成器模型和判别器模型，其中，在训练过程中利用至少一个第一人群图像对所述生成对抗网络反复执行如下操作，直到所述判别器模型判别出的相似性满足预设阈值条件，包括：

基于对所述第一人群图像中的人群标定和变换处理，获得所述第一人群密度真值图；

以第一人群图像作为所述生成器模型的输入，训练所述生成器模型输出第一人群密度估计图；以及

在训练过程中以所述第一人群密度真值图和所述第一人群密度估计图作为所述判别器模型的输入，以使所述判别器模型判别所述第一人群密度真值图和所述第一人群密度估计图的相似性；

人数估计模块，用于在训练结束后，以目标监控区域的第二人群图像作为所述生成器模型的输入，基于所述生成器模型输出的第二人群密度估计图来估计所述目标监控区域中的人数。

9.一种电子设备，包括：

一个或多个处理器；

存储器，用于存储一个或多个程序，

其中，当所述一个或多个程序被所述一个或多个处理器执行时，使得所述一个或多个处理器执行根据权利要求1～7中任一项所述的方法。

10.一种计算机可读存储介质，其上存储有可执行指令，该指令被处理器执行时使处理器执行根据权利要求1～7中任一项所述的方法。