基于NonshadowGAN的无人机铁路扣件图像阴影去除方法及系统

    公开(公告)号:CN114266713A

    公开(公告)日:2022-04-01

    申请号:CN202111615470.5

    申请日:2021-12-27

    Abstract: 本申请提供一种基于NonshadowGAN的无人机铁路扣件图像阴影去除方法及系统。本申请基于无人机所获取的铁路扣件图像,首先对扣件图像进行预处理实现数据的增广,然后利用NonshadowGAN网络实现对扣件部件的阴影去除。本申请将无人机铁路扣件图像输入到阴影去除网络中时,将损失函数分为三部分,通过生成对抗网络将对抗损失应用于学习阴影特征,完成阴影和无阴影特征的学习与识别任务,并通过循环一致性损失减少映射函数的可用空间,通过身份验证损失进一步约束网络结构,鉴别阴影生成结果,调整参数模型。从而,本申请能够基于非成对的阴影扣件数据实现模型训练,进而准确实现对无人机扣件数据中阴影部分的去除。

Patent Agency Ranking