一种基于机器学习的公交车乘客量估算方法

摘要

本发明涉及一种基于机器学习的公交车乘客量估算方法，本发明属于机器学习和智能交通领域。本发明首先通过在公交车内架设摄像头，获取一定量视频图像作为训练样本；然后对样本进行标注，计算每个样本的ROI概率密度分布图；然后将样本和对应的ROI概率密度分布图送至深度卷积神经网络进行训练，训练生成模型；最后获取一张待检测视频图像，使用训练生成的模型回归该图像的ROI概率密度分布图，将ROI概率密度分布图所有像素累加，该累加值近似为该待检测图像中乘客量。本发明部署简单，时间开销大部分集中在机器学习阶段，实际使用时，直接使用训练好的模型即可，估算乘客量的计算量很小，计算速度快。