[发明专利]一种公交车乘客OD匹配方法、装置和电子设备

权利要求书

1.一种公交车乘客OD匹配方法，该方法包括：

获取每个上车乘客在上车前的连续n帧第一截图，以及每个下车乘客在下车前的连续m帧第二截图；其中，n、m为大于1的整数；

将所述n帧第一截图和所述m帧第二截图分别输入预先训练好的ReID模型，以得到对应每个上车乘客的n个第一向量和对应每个下车乘客的m个第二向量；

根据每个上车乘客的n个第一向量和每个下车乘客的m个第二向量，对上车乘客和下车乘客进行匹配；

其中，根据每个上车乘客的n个第一向量和每个下车乘客的m个第二向量，对上车乘客和下车乘客进行匹配，包括：

针对任一上车乘客和任一下车乘客，分别计算该上车乘客的每个第一向量和该下车乘客的每个第二向量的余弦相似度，以得到每个上车乘客与每个下车乘客的n*m个余弦相似度；

计算n*m个余弦相似度的平均值，以得到每个上车乘客与每个下车乘客的余弦相似度平均值；

基于该余弦相似度平均值对上车乘客和下车乘客进行匹配。

2.根据权利要求1所述的方法，在获取每个上车乘客在上车前的连续n帧第一截图，以及每个下车乘客在下车前的连续m帧第二截图，之前，所述方法还包括：

获取各时间段的乘客上下车的视频数据；

针对任一乘客，根据所述视频数据判断该乘客在对应时间段为上车乘客还是下车乘客；

根据判断结果，得到各时间段的上车乘客视频数据和下车乘客视频数据。

3.根据权利要求2所述的方法，根据所述视频数据判断该乘客在对应时间段为上车乘客还是下车乘客，包括：

针对任一乘客，利用预先训练好的乘客检测与跟踪模型对该乘客进行检测与跟踪，并利用跟踪框提取该乘客的轨迹；

根据该乘客的轨迹判断该乘客为上车乘客还是下车乘客。

4.根据权利要求3所述的方法，根据该乘客的轨迹判断该乘客为上车乘客还是下车乘客，包括：

基于该乘客的轨迹，记录该乘客跟踪框的初始位置和终点位置；

若该乘客的初始位置位于公交车的车门区域外，且终点位置位于公交车的车门区域内，则判断该乘客为下车乘客；

若该乘客的初始位置位于公交车的车门区域内，且终点位置位于公交车的车门区域外，则判断该乘客为上车乘客。

5.根据权利要求4所述的方法，获取每个上车乘客在上车前的连续n帧第一截图，以及每个下车乘客在下车前的连续m帧第二截图，包括：

针对任一乘客，每隔预设时间连续截取乘客检测与跟踪模型对该乘客的跟踪框；

基于该乘客跟踪框的初始位置和终点位置，当判断该乘客为上车乘客或下车乘客时，获取该乘客在上车前的连续n帧第一截图或下车前的连续m帧第二截图。

6.根据权利要求1所述的方法，在计算该上车乘客的每个第一向量和该下车乘客的每个第二向量的余弦相似度之前，该方法还包括：

获取该上车乘客的上车时间和该下车乘客的下车时间；

若该下车乘客的下车时间早于该上车乘客的上车时间，则将判断该下车乘客与该上车乘客不匹配；

否则，再计算该上车乘客的每个第一向量和该下车乘客的每个第二向量的余弦相似度。

7.根据权利要求1所述的方法，基于该余弦相似度平均值对上车乘客和下车乘客进行匹配，包括：

基于该余弦相似度平均值，利用匈牙利算法对上车乘客和下车乘客进行匹配。

8.根据权利要求2所述的方法，获取各时间段的乘客上下车的视频数据，具体包括：

利用设置在公交车前门区域和后门区域的摄像头分别获取各时间段的乘客上下车的视频数据。