摘要
本发明涉及一种模型的交通灯的识别方法、训练方法、设备及车辆,包括获取交通灯识别训练集;其中,交通灯识别训练集中包括多个携带有类别标签的历史交通灯图片;基于所获取的交通灯识别训练集中包括的多个携带有类别标签的历史交通灯图片,对初始交通灯识别模型中包括的主干网络、基础分类网络、噪声生成网络以及噪声判别网络进行对抗训练,得到交通灯识别模型。本发明在基于噪声生成网络进行的对抗训练可以提高训练得到的交通灯识别模型的鲁棒性,在此基础上增加噪声判别网络可以避免训练过程中对引入的噪声过拟合,提高训练得到的交通灯识别模型的准确性。
技术关键词
交通灯识别
分类网络
图片
噪声
标签
样本
计算机执行指令
上采样
基础
训练集
识别方法
编码
模型训练装置
参数
模型训练方法
颜色
处理器
可读存储介质