基于改进CycleGAN的路面阴影数据增广方法
摘要:
本发明提出了一种基于改进CycleGAN的路面阴影数据增广方法,包括以下步骤:步骤S100:获取路面数据集,包括无阴影集A和有阴影集B两类,两类数据集均包括训练集train和验证集val;步骤S200:搭建现有的CycleGAN,之后增加纹理自监督机制Losstss以形成改进的CycleGAN,以保留原本图像的纹理信息;步骤S300:设置损失函数和训练参数,之后基于训练集train对步骤S200输出的模型进行训练,最后基于验证集val验证;步骤S400:建立无阴影路面图片的待处理数据集,通过训练好的生成器G_A2B将无阴影数据生成有阴影数据。本发明生成含有阴影的路面数据集,解决了阴影路面缺乏的问题。
0/0