基于生成式对抗网络的城市交通流量预测方法

摘要

本发明提供了一种基于生成式对抗网络的城市交通流量预测方法，其包括如下步骤：获取某一城市的位置信息并将该城市区域划分为多个网格区域；采集预设时间内该城市区域中车辆的GPS轨迹数据集，根据时间段统计计算出每个网格区域中的进流量和出流量，并将进流量和出流量组织成流量图集I；将连续的三帧流量图I_N‑2、I_N‑1、I_N输入到训练好的生成式对抗网络模型中，即可得到预测的下一个时间间隔的流量图I_N+1，其中N大于2。与相关技术相比，本发明的基于生成式对抗网络的城市交通流量预测方法采用3D卷积能够有效提取交通流量的时间强相关性和空间依赖性，采用生成流量图的方式来预测，预测结果可靠性更高。