基于Faster R-CNN的车辆多属性识别

Vehicle Multi-attribute Recognition Based on Faster R-CNN

基于Faster R-CNN提出一种车辆的多属性识别模型。首先利用Faster R-CNN对车辆数据库进行训练,得到车辆检测网络,对图像中多个车辆目标进行检测。将检测结果输入改进的车辆属性识别网络中,对检测得到的车辆进行属性推断,包括车辆颜色、品牌和姿态。为评估车辆检测精度和车辆多属性识别的准确率,采集了8 000张真实场景下的图片作为测试集进行测试。对于车辆检测网络,对比了R-CNN、Fast R-CNN等方法的检测精度;对于车辆属性识别,对比了不同网络结构、不同图片分辨率和单属性和多属性等对于识别准确率的影响。实验结果表明,基于Faster R-CNN的车辆多属性识别方法充分学习了不同属性间的特征,具有较高的准确率和检测精度,以及良好的通用性和鲁棒性,适用于车辆多属性分类。

We put forward a vehicle multi-attribute recognition model based on Faster R-CNN. First we use vehicle images to train andobtain vehicle detection network for detection of vehicle targets in image. Then we put detected results into vehicle attribute recognitionnetwork and infer attributes including color,type and view. In order to evaluate the accuracy of vehicle detection precision and vehiclemulti-attributes detection,we collect 8 000 vehicle images under actual scene as test set for testing. In terms of vehicle detection network,we compare the detection precision of R-CNN and Fast R-CNN,and for vehicle attribute recognition we compare the accuracy ofdifferent network,different image definition and different number of attributes. Experiment shows that the proposed vehicle multi-attribute recognition method based on Faster R-CNN can learn more features fully with higher accuracy and precision,as well as better versatility and robustness,which can be used for vehicle multi-attribute classification.