发明公开
- 专利标题: 基于改进CycleGAN的路面阴影数据增广方法
-
申请号: CN202410200004.8申请日: 2024-02-23
-
公开(公告)号: CN118097330A公开(公告)日: 2024-05-28
- 发明人: 吴光强 , 毛瑞驰 , 王浩 , 颜旭东 , 杨雨辰 , 鞠丽娟
- 申请人: 同济大学
- 申请人地址: 上海市杨浦区四平路1239号
- 专利权人: 同济大学
- 当前专利权人: 同济大学
- 当前专利权人地址: 上海市杨浦区四平路1239号
- 代理机构: 上海邦德专利代理事务所
- 代理商 刘旭章
- 主分类号: G06V10/774
- IPC分类号: G06V10/774 ; G06V10/82 ; G06N3/045 ; G06N3/0475 ; G06N3/084 ; G06N3/094
摘要:
本发明提出了一种基于改进CycleGAN的路面阴影数据增广方法,包括以下步骤:步骤S100:获取路面数据集,包括无阴影集A和有阴影集B两类,两类数据集均包括训练集train和验证集val;步骤S200:搭建现有的CycleGAN,之后增加纹理自监督机制Losstss以形成改进的CycleGAN,以保留原本图像的纹理信息;步骤S300:设置损失函数和训练参数,之后基于训练集train对步骤S200输出的模型进行训练,最后基于验证集val验证;步骤S400:建立无阴影路面图片的待处理数据集,通过训练好的生成器G_A2B将无阴影数据生成有阴影数据。本发明生成含有阴影的路面数据集,解决了阴影路面缺乏的问题。