基于注意力机制优化的轻量化车辆车牌颜色识别方法

专利主权项内容

1.基于注意力机制优化的轻量化车辆车牌颜色识别方法，其特征在于，包括以下步骤：S1.获取车辆行驶数据，在车辆行驶数据中获得车辆图像，并标注车辆车牌位置；S2.根据车辆车牌颜色为车辆车牌位置的标注框设定标签；S3.对车辆图像进行分辨率统一处理，形成训练数据集；S4.构建车辆车牌颜色识别网络结构，生成车辆车牌颜色识别模型，并对模型训练；所述车辆车牌颜色识别模型网络结构包括CBS模块、C3模块、SlimAM模块、Upsample模块和Concat模块；CBS模块包括卷积层、批量归一化层和SiLU激活函数；其中，卷积层的卷积核大小为3x3；C3模块包括第一分支和第二分支，第一分支和第二分支进行Concat操作后输出；所述第一分支包括CSP、CBS模块，CBS模块的卷积核大小为1x1，步长为1；所述第二分支为CBS模块，卷积核大小为1x1，步长为1；所述CSP包括两个CBS模块；其中，一个CBS模块的卷积核大小为1x1，步长为1，另一个CBS模块的卷积核大小为3x3，步长为1；S5.将车辆图像数据输入至车辆车牌颜色识别中，输出车牌颜色。