基于强化学习图像亮度调节的空中视角人群计数方法

摘要

本发明公开了一种基于强化学习图像亮度调节的空中视角人群计数方法，首先获取经标注的训练集，对其预处理，得到原始分辨率图像块和低分辨率图像块的标注数据集；基于得到的数据集训练得到原始人群计数模型和粗糙人群计数模型；然后构建并训练强化策略网络模型。工作时，对于待处理的无人机实时图像，利用训练得到的强化策略网络模型选取对应分辨率的图像块并进行亮度值的调整，根据选择结果调取对应的人群计数模型得到各图像块的人群计数结果，综合得到整张图像的人群计数结果。本发明不仅能够在只有空中视觉信息的条件下实现人群计数，还能够基于训练的强化学习模型对原始模型进行加速。