万斌杨, 侯进. 基于EfficientNetV2模型迁移的车辆图像识别算法[J]. 微电子学与计算机, 2022, 39(10): 62-70. DOI: 10.19304/J.ISSN1000-7180.2022.0150
引用本文: 万斌杨, 侯进. 基于EfficientNetV2模型迁移的车辆图像识别算法[J]. 微电子学与计算机, 2022, 39(10): 62-70. DOI: 10.19304/J.ISSN1000-7180.2022.0150
WAN Binyang, HOU Jin. Vehicle image recognition algorithm based on EfficientNetV2 model transfer learning[J]. Microelectronics & Computer, 2022, 39(10): 62-70. DOI: 10.19304/J.ISSN1000-7180.2022.0150
Citation: WAN Binyang, HOU Jin. Vehicle image recognition algorithm based on EfficientNetV2 model transfer learning[J]. Microelectronics & Computer, 2022, 39(10): 62-70. DOI: 10.19304/J.ISSN1000-7180.2022.0150

基于EfficientNetV2模型迁移的车辆图像识别算法

Vehicle image recognition algorithm based on EfficientNetV2 model transfer learning

  • 摘要: 智能交通系统中,由于车辆流动性大、道路拍摄环境恶劣等问题,传统深度学习大多只能识别车牌与数量,该领域数据集也极为稀少,基于这些问题提出一种EfficientNetV2为主干网络的图像识别模型.使用ImageNet的预训练参数,冻结部分网络快速提取图像特征,选取VOC2012数据集中交通工具部分进行领域适配,得到一个可移植性强的交通工具识别模型.将上一步得到的模型参数作为源域,二次迁移至车辆数据集,与ImageNet作为源域的单次迁移对比可知,领域相似性越高的领域迁移效果更好.最后, 以上一步获得的特征提取网络及相应参数为基础, 结合子空间变换法Coral对不同网络深度的特征进行约束,令模型适配特征分布有差异的新任务并加快收敛, 并以随机抽取的少量样本检测模型是否有过拟合现象.通过实验可知,在小样本数据集下运用迁移学习后模型的识别精度和训练速度大幅提升,并且能便捷地再次使用于其它相似领域的识别任务.

     

    Abstract: In intelligent transportation system, due to the problems of large vehicle mobility and poor road shooting environment, most of the traditional deep learning can only recognize the number of vehicles and license plates, and the data set in this field is also extremely scarce. Based on these problems, an image recognition model with EfficientNetV2 as the backbone network is proposed. Using the pre training parameters of ImageNet, and freezing part of the network to quickly extract image features. Select the transportation facility related parts in VOC2012 data set for domain adaptation training, and get a vehicle recognition model with strong portability. Take the model parameters obtained in the previous step as the source domain, and transfer the parameters to the vehicle data set again. Compared with the single transfer of ImageNet as the source domain, it can be seen that the domain with higher domain similarity has better transfer effect. Finally, based on the feature extraction network and corresponding parameters obtained in the previous step, combined with the subspace transformation method Coral, the features of different network depths are constrained, so that the model can be adapted to new tasks with different feature distributions and accelerate convergence. A small number of samples are used to detect whether the model is overfitting. Through experiments, it can be seen that the recognition accuracy and training speed of the model are greatly improved after transfer learning is used in a small sample data set, and it can be easily reused for recognition tasks in other similar fields.

     

/

返回文章
返回