一种单通道接触网暗光图像增强方法

    公开(公告)号:CN118247189A

    公开(公告)日:2024-06-25

    申请号:CN202410146976.3

    申请日:2024-02-02

    摘要: 本发明提供一种单通道接触网暗光图像增强方法,包括:步骤一、对铁路接触网进行图像采集获得其三通道RGB图并建立接触网暗光图像数据集;步骤二、建立接触网暗光图像数据集中图像与正常效果图像的配对数据集;将其接触网三通道RGB图转换为单通道灰度图,并按9:1划分成训练集和测试集;构建深度强化学习模型,其包括智能体、动作、状态和奖励四个模块;步骤三、将训练集和测试集输入深度强化学习模型进行学习,获得增强后的接触网图像。应用本发明的方法,通过改进深度强化学习网络的动作组合方式对接触网低亮度图像进行增强处理,提高图像质量,改变传统的全局性图像增强策略,克服了传统全局化增强方法受到图像大范围黑色背景干扰的缺陷。