一种基于图像风格迁移的交通信号灯图像转换方法及系统

摘要

本发明涉及一种基于图像风格迁移的交通信号灯图像转换方法及系统，其方法包括：获取多张包含第一交通信号灯源图像，以及至少一张包含第二交通信号灯的目标图像；利用训练完成的CycleGAN，将每张源图像中的第一交通信号灯转换为第二信号灯，通过前景提取方法分别提取转换后的每张源图像中的第二交通信号灯；通过图像融合方法将提取到的多个第二交通信号灯，分别替换掉目标图像中的第一交通信号灯。本发明通过CycleGAN将样本量大的交通信号灯图像转换为样本量小的交通信号灯图像，再通过前景提取和图像融合算法，增广了交通信号灯的样本量，提高了转换后的图像质量。