摘要
本发明公开了一种基于CycleGAN的晴雪天气图像数据风格迁移系统,包括:网络构建模块构建基于CycleGAN的晴雪天气图像数据风格迁移模型;掩码模块根据输入图像的图像注释信息创建掩码,掩码与输入图像叠加后输入教师网络与学生网络边缘检测;输入图像经第一生成器进行风格迁移得到生成图像,生成图像与掩码叠加后输入学生网络边缘检测;生成图像经第二生成器生成重建图像,重建图像与掩码叠加输入学生网络边缘检测;模型训练模块对构建的晴雪天气图像数据风格迁移模型进行训练;推理模块接收晴朗天气图像,生成对应的雪天图像。本发明能够解决现有风格迁移技术在晴雪天气图像数据风格迁移中存在的对象丢失的技术问题。
技术关键词
风格迁移方法
迁移系统
天气
边缘检测器
网络
模型训练模块
学生
教师
交通
数据
蒸馏
风格迁移技术
像素
图像边缘检测
列表
通道