Journal of Computer Applications ›› 2023, Vol. 43 ›› Issue (4): 1291-1296.DOI: 10.11772/j.issn.1001-9081.2022020313
• Multimedia computing and computer simulation • Previous Articles
Zhouhua ZHU, Qi QI()
Received:
2022-03-18
Revised:
2022-07-14
Accepted:
2022-07-18
Online:
2022-08-16
Published:
2023-04-10
Contact:
Qi QI
About author:
ZHU Zhouhua, born in 1976, M. S., associate professor. Her research interests include machine learning, object detection, digital signal processing.
Supported by:
通讯作者:
齐琦
作者简介:
朱周华(1976—),女,陕西西安人,副教授,硕士,主要研究方向:机器学习、目标检测、数字信号处理;
基金资助:
CLC Number:
Zhouhua ZHU, Qi QI. Automatic detection and recognition of electric vehicle helmet based on improved YOLOv5s[J]. Journal of Computer Applications, 2023, 43(4): 1291-1296.
朱周华, 齐琦. 基于改进YOLOv5s电动车头盔的自动检测与识别[J]. 《计算机应用》唯一官方网站, 2023, 43(4): 1291-1296.
Add to citation manager EndNote|Ris|BibTeX
URL: http://www.joca.cn/EN/10.11772/j.issn.1001-9081.2022020313
类别 | 精度 | 召回率 | mAP0.5 | |||
---|---|---|---|---|---|---|
改进前 | 改进后 | 改进前 | 改进后 | 改进前 | 改进后 | |
平均值 | 79.3 | 88.2 | 83.7 | 85.3 | 84.2 | 91.3 |
electric | 84.2 | 91.6 | 91.5 | 91.5 | 93.1 | 96.9 |
helmet | 86.2 | 90.5 | 88.5 | 92.6 | 91.1 | 96.2 |
no helmet | 67.5 | 82.4 | 71.1 | 71.8 | 68.4 | 80.8 |
Tab. 1 Performance comparison of YOLOv5s before and after improvement
类别 | 精度 | 召回率 | mAP0.5 | |||
---|---|---|---|---|---|---|
改进前 | 改进后 | 改进前 | 改进后 | 改进前 | 改进后 | |
平均值 | 79.3 | 88.2 | 83.7 | 85.3 | 84.2 | 91.3 |
electric | 84.2 | 91.6 | 91.5 | 91.5 | 93.1 | 96.9 |
helmet | 86.2 | 90.5 | 88.5 | 92.6 | 91.1 | 96.2 |
no helmet | 67.5 | 82.4 | 71.1 | 71.8 | 68.4 | 80.8 |
算法 | 精度 | 召回率 | mAP0.5 |
---|---|---|---|
文献[ | 96.0 | 73.0 | 85.0 |
本文算法 | 88.2 | 86.4 | 93.1 |
Tab. 2 Performance comparison of different algorithms
算法 | 精度 | 召回率 | mAP0.5 |
---|---|---|---|
文献[ | 96.0 | 73.0 | 85.0 |
本文算法 | 88.2 | 86.4 | 93.1 |
模型 | 精度 | 召回率 | mAP0.5 |
---|---|---|---|
原始YOLOv5s | 79.3 | 83.7 | 84.2 |
YOLOv5s+注意力机制 | 81.6 | 84.5 | 86.8 |
YOLOv5s+DIoU-NMS | 80.1 | 84.1 | 86.4 |
YOLOv5s+多尺度特征融合 | 80.8 | 83.9 | 87.6 |
Tab. 3 YOLOv5s ablation experiment results
模型 | 精度 | 召回率 | mAP0.5 |
---|---|---|---|
原始YOLOv5s | 79.3 | 83.7 | 84.2 |
YOLOv5s+注意力机制 | 81.6 | 84.5 | 86.8 |
YOLOv5s+DIoU-NMS | 80.1 | 84.1 | 86.4 |
YOLOv5s+多尺度特征融合 | 80.8 | 83.9 | 87.6 |
算法 | 精度 | 召回率 | mAP0.5 |
---|---|---|---|
YOLOv5s | 79.3 | 83.7 | 84.2 |
YOLOv5m | 83.6 | 85.6 | 88.2 |
YOLOv5l | 84.8 | 86.1 | 90.1 |
YOLOv5x | 84.1 | 85.5 | 89.4 |
改进YOLOv5s | 88.2 | 86.4 | 91.3 |
Tab. 4 Performance comparison between the proposed algorithm and YOLOv5 series algorithms
算法 | 精度 | 召回率 | mAP0.5 |
---|---|---|---|
YOLOv5s | 79.3 | 83.7 | 84.2 |
YOLOv5m | 83.6 | 85.6 | 88.2 |
YOLOv5l | 84.8 | 86.1 | 90.1 |
YOLOv5x | 84.1 | 85.5 | 89.4 |
改进YOLOv5s | 88.2 | 86.4 | 91.3 |
1 | REN S Q, HE K M, GIRSHICK R, et al. Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, 39(6):1137-1149. 10.1109/tpami.2016.2577031 |
2 | HE K M, GKIOXARI G, DOLLÁR P, et al. Mask R-CNN[C]// Proceedings of the 2017 IEEE International Conference on Computer Vision. Piscataway: IEEE, 2017: 2980-2988. 10.1109/iccv.2017.322 |
3 | REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: unified real-time object detection[C]// Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE, 2016:779-788. 10.1109/cvpr.2016.91 |
4 | LIU W, ANGUELOV D, ERHAN D, et al. SSD: single shot multibox detector[C]// Proceedings of the 2014 European Conference on Computer Vision, LNCS 9905. Cham: Springer, 2016:21-37. |
5 | LIN T Y, GOYAL P, GIRSHICK R, et al. Focal loss for dense object detection[C]// Proceedings of the 2017 IEEE International Conference on Computer Vision. Piscataway: IEEE, 2017: 2999-3007. 10.1109/iccv.2017.324 |
6 | ZHANG S, CHI C, YAO Y, et al. Bridging the gap between anchor-based and anchor-free detection via adaptive training sample selection[C]// Proceedings of the 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE, 2020:9756-9765. 10.1109/cvpr42600.2020.00978 |
7 | YANG Z, LIU S, HU H, et al. RepPoints: point set representation for object detection[C]// Proceedings of the 2019 IEEE/CVF International Conference on Computer Vision. Piscataway: IEEE, 2019:9656-9665. 10.1109/iccv.2019.00975 |
8 | TIAN Z, SHEN C H, CHEN H, et al. FCOS: fully convolutional one-stage object detection[C]// Proceedings of the 2019 IEEE/CVF International Conference on Computer Vision. Piscataway: IEEE, 2019:9626-9635. 10.1109/iccv.2019.00972 |
9 | BHAGAT S, CONTRACTOR D, SHARMA S, et al. Cascade classifier based helmet detection using OpenCV in image processing[C/OL]// Proceedings of the 2016 National Conference on Recent Trends in Computer and Communication Technology. [2021-02-28].. |
10 | SILVA R, AIRES K, SANTOS T, et al. Automatic detection of motorcyclists without helmet[C]// Proceedings of the XXXIX Latin American Computing Conference. Piscataway: IEEE, 2013:1-7. 10.1109/clei.2013.6670613 |
11 | YOGAMEENA B, MENAKA K, PERUMAAL S S. Deep learning-based helmet wear analysis of a motorcycle rider for intelligent surveillance system[J]. IET Intelligent Transport Systems, 2019, 13(7):1190-1198. 10.1049/iet-its.2018.5241 |
12 | VISHNU C, SINGH D, MOHAN C K, et al. Detection of motorcyclists without helmet in videos using convolutional neural network[C]// Proceedings of the 2017 International Joint Conference on Neural Networks. Piscataway: IEEE, 2017:3036-3041. 10.1109/ijcnn.2017.7966233 |
13 | SHINE L, JIJI C V. Automated detection of helmet on motorcyclists from traffic surveillance videos: a comparative analysis using hand-crafted features and CNN[J]. Multimedia Tools Applications, 2020, 79(19/20): 14179-14199. 10.1007/s11042-020-08627-w |
14 | CHAIRAT A, DAILEY M N, LIMSOONTHRAKUL S, et al. Low cost, high performance automatic motorcycle helmet violation detection[C]// Proceedings of the 2020 IEEE Winter Conference on Applications of Computer Vision. Piscataway: IEEE, 2020:3549-3557. 10.1109/wacv45572.2020.9093538 |
15 | SINGH D, VISHNU C, MOHAN C K. Real-time detection of motorcyclist without helmet using cascade of CNNs on edge-device[C]// Proceedings of the IEEE 23rd International Conference on Intelligent Transportation Systems. Piscataway: IEEE, 2020:1-8. 10.1109/itsc45102.2020.9294747 |
16 | DASGUPTA M, BANDYOPADHYAY O, CHATTERJI S. Automated helmet detection for multiple motorcycle riders using CNN[C]// Proceedings of the 2019 IEEE Conference on Information and Communication Technology. Piscataway: IEEE, 2019:1-4. 10.1109/cict48419.2019.9066191 |
17 | 赵睿,刘辉,刘沛霖,等. 基于改进YOLOv5s的安全帽检测算法[J/OL]. 北京航空航天大学学报 (2021-11-23) [2022-01-23].. 10.1109/icccas55266.2022.9825037 |
ZHAO R, LIU H, LIU P L, et al. Research on helmet detection algorithm based on improved YOLOv 5s[J/OL]. Journal of Beijing University of Aeronautics and Astronautics (2021-11-23) [2022-01-23].. 10.1109/icccas55266.2022.9825037 | |
18 | HE K M, ZHANG X Y, REN S Q, et al. Spatial pyramid pooling in deep convolutional networks for visual recognition[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2015, 37(9):1904-1916. 10.1109/tpami.2015.2389824 |
19 | LIN T Y, DOLLÁR P, GIRSHICK R, et al. Feature pyramid networks for object detection[C]// Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE, 2017: 936-944. 10.1109/cvpr.2017.106 |
20 | LIU S, QI L, QIN H F, et al. Path aggregation network for instance segmentation[C]// Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE, 2018:8759-8768. 10.1109/cvpr.2018.00913 |
21 | WOO S, PARK J, LEE J Y, et al. CBAM: convolutional block attention module[C]// Proceedings of the 2018 European Conference on Computer Vision, LNCS 11211. Cham: Springer, 2018: 3-19. |
22 | HOU Q B, ZHOU D Q, FENG J S. Coordinate attention for efficient mobile network design[C]// Proceedings of the 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE, 2021:13708-13717. 10.1109/cvpr46437.2021.01350 |
23 | 邹梓吟,盖绍彦,达飞鹏,等. 基于注意力机制的遮挡行人检测算法[J]. 光学学报, 2021, 41(15): No.1515001. 10.3788/aos202141.1515001 |
ZOU Z Y, GAI S Y, DA F P, et al. Occluded pedestrian detection algorithm based on attention mechanism[J]. Acta Optica Sinica, 2021, 41(15): No.1515001. 10.3788/aos202141.1515001 | |
24 | WU W K, ZHANG Y, WANG D, et al. SK-Net: deep learning on point cloud via end-to-end discovery of spatial keypoints[C]// Proceedings of the 34th AAAI Conference on Artificial Intelligence. Palo Alto, CA: AAAI Press, 2020: 6422-6429. 10.1609/aaai.v34i04.6113 |
25 | ZHANG Z H, WANG P, LIU W, et al. Distance-IoU loss: faster and better learning for bounding box regression[C]// Proceedings of the 34th AAAI Conference on Artificial Intelligence. Palo Alto, CA: AAAI Press, 2020:12993-13000. 10.1609/aaai.v34i07.6999 |
26 | LI S S, LI Y J, LI Y, et al. YOLO-FIRI: improved YOLOv5 for infrared image object detection[J]. IEEE Access, 2021, 9: 141861-141875. 10.1109/access.2021.3120870 |
27 | IOFFE S, SZEGEDY C. Batch normalization: accelerating deep network training by reducing internal covariate shift[C]// Proceedings of the 32nd International Conference on Machine Learning. New York: JMLR.org, 2015:448-456. |
28 | 樊缤,李智,高健. 基于多尺度知识学习的深度鲁棒水印算法[J]. 计算机应用, 2022, 42(10):3102-3110. 10.11772/j.issn.1001-9081.2021050737 |
FAN B, LI Z, GAO J. Deep robust watermarking algorithm based on multiscale knowledge learning[J]. Journal of Computer Applications, 2022, 42(10):3102-3110. 10.11772/j.issn.1001-9081.2021050737 | |
29 | 姚群力,胡显,雷宏. 基于多尺度融合特征卷积神经网络的遥感图像飞机目标检测[J]. 测绘学报, 2019, 48(10):1266-1274. 10.11947/j.AGCS.2019.20180398 |
YAO Q L, HU X, LEI H. Aircraft detection in remote sensing imagery with multi-scale feature fusion convolutional neural networks[J]. Acta Geodaetica et Cartographica Sinica, 2019, 48(10):1266-1274. 10.11947/j.AGCS.2019.20180398 | |
30 | 左航旭,廖彬,陈小昆,等. 融合迁移学习和数据增强的SC-Net模型在皮肤癌识别中的应用[J]. 计算机应用研究, 2022, 39(8):2550-2555, 2560. |
ZUO H X, LIAO B, CHEN X K, et al. Application of SC-Net model integrated with transfer learning and data augmentation in skin cancer recognition[J]. Application Research of Computers, 2022, 39(8):2550-2555, 2560. |
[1] | Juming HAO, Jingyu YANG, Shumei HAN, Yangping WANG. YOLOv4 highway pavement crack detection method using Ghost module and ECA [J]. Journal of Computer Applications, 2023, 43(4): 1284-1290. |
[2] | Quan YUAN, Yunpeng XU, Chengliang TANG. Document-level relation extraction method based on path labels [J]. Journal of Computer Applications, 2023, 43(4): 1029-1035. |
[3] | Hao SUN, Jian CAO, Haisheng LI, Dianhui MAO. Session-based recommendation model based on enhanced capsule network [J]. Journal of Computer Applications, 2023, 43(4): 1043-1049. |
[4] | Jiazhen ZU, Yongxia ZHOU, Le CHEN. Dual-branch residual low-light image enhancement combined with attention [J]. Journal of Computer Applications, 2023, 43(4): 1240-1247. |
[5] | Guangyi DOU, Fanan WEI, Chuangyi QIU, Jianshu CHAO. Tracking appearance features based on attention self-correlation mechanism [J]. Journal of Computer Applications, 2023, 43(4): 1248-1254. |
[6] | Jiadong LI, Danpu ZHANG, Yaqiong FAN, Jianfeng YANG. Lightweight ship target detection algorithm based on improved YOLOv5 [J]. Journal of Computer Applications, 2023, 43(3): 923-929. |
[7] | Cong YIN, Hanping HU. Parameter identification model for time-delay chaotic systems based on temporal attention mechanism [J]. Journal of Computer Applications, 2023, 43(3): 842-847. |
[8] | Jie SUN, Shaoxin WU, Xuejun WANG, Jing HUA. Efficient person search algorithm and optimization with Sophon SC5+ chip architecture [J]. Journal of Computer Applications, 2023, 43(3): 744-751. |
[9] | Xuedong HE, Shibin XUAN, Kuan WANG, Mengnan CHEN. DeepLabV3+ image segmentation algorithm fusing cumulative distribution function and channel attention mechanism [J]. Journal of Computer Applications, 2023, 43(3): 936-942. |
[10] | Cong LIU, Genshun WAN, Jianqing GAO, Zhonghua FU. End-to-end speech recognition method based on prosodic features [J]. Journal of Computer Applications, 2023, 43(2): 380-384. |
[11] | Xiaomeng SHAO, Meng ZHANG. Temporal convolutional knowledge tracing model with attention mechanism [J]. Journal of Computer Applications, 2023, 43(2): 343-348. |
[12] | Guihui CHEN, Jinyu LIN, Yuehua LI, Zhongbing LI, Yuli WEI, Kai LU. Multi-stage low-illuminance image enhancement network based on attention mechanism [J]. Journal of Computer Applications, 2023, 43(2): 552-559. |
[13] | Ming XU, Linhao LI, Qiaoling QI, Liqin WANG. Abductive reasoning model based on attention balance list [J]. Journal of Computer Applications, 2023, 43(2): 349-355. |
[14] | Ping WANG, Nan CHEN, Lei LU. Fall detection algorithm based on scene prior and attention guidance [J]. Journal of Computer Applications, 2023, 43(2): 529-535. |
[15] | Zihao GUO, Lele DONG, Zhijian QU. Arthropod object detection method based on improved Faster RCNN [J]. Journal of Computer Applications, 2023, 43(1): 88-97. |
Viewed | ||||||
Full text |
|
|||||
Abstract |
|
|||||