Computer Science ›› 2022, Vol. 49 ›› Issue (4): 239-246.doi: 10.11896/jsjkx.210200119

• Computer Graphics & Multimedia • Previous Articles     Next Articles

Study on Reflective Vest Detection for Apron Workers Based on Improved YOLOv3 Algorithm

XU Tao, CHEN Yi-ren, LYU Zong-lei   

  1. School of Computer Science and Technology, Civil Aviation University of China, Tianjin 300000, China
  • Received:2021-02-19 Revised:2021-07-02 Published:2022-04-01
  • About author:XU Tao,born in 1962,professor,is a member of China Computer Federation.His main research interests include intelligent information processing and so on.LYU Zong-lei,born in 1981,Ph.D,asso-ciate professor,is a member of China Computer Federation.His main research interests include machine lear-ning and so on.
  • Supported by:
    This work was supported by the Fundamental Research Funds for Central Universities of the Civil Aviation University of China(3122021088).

Abstract: This paper proposes a reflective vest detection algorithm for apron staff based on prior knowledge and improved YOLOv3 algorithm.Aiming at the problem of the existing target detection method with low speed, the reflective vest detection candidate region is generated based on prior knowledge to replace the initial candidate region, so as to reduce the detection area.Darknet-37 is used to replace Darknet-53 as the backbone network for feature extraction, which improves the detection speed of the algorithm.Aiming at the problem that the reflective vest occupies a small area in the picture and is difficult to identify, a spatial pyramid pooling structure (SPP) is added into the detection model to realize feature enhancement, and the detection scale is increased to four for multi-scale feature fusion.The K-means++algorithm is used to perform cluster analysis again on the size of labeled bounding box, and the clustering result is used to replace the initial Anchor value of Yolov3.GIoU is selected as the loss function to improve the positioning accuracy.Experimental results show that the proposed new target detection algorithm in the self-built reflective vest data set is better than YOLOv3 test results, the precision rate and recall rate reach 97.6% and 96.1%, detection rate reach 28.4 frames/s, which effectively solves the problems such as inaccurate positioning, missed detection and low detection speed existing in the original model, and meets the real-time requirements in the practical application of apron target detection while ensuring a high detection accuracy.

Key words: Feature fusion, Object detection, Real-time detection, Reflective vest detection, Spatial pyramid pooling

CLC Number: 

  • TP391.4
[1] ZHAO Y Q,RAO Y,DONG S P,et al.Survey on deep learning object detection[J].Journal of Image and Graphics,2020,25(4):629-654.
[2] GIRSHICK R.Fast R-CNN[C]//Proceedings of the IEEE International Conference on Computer Vision,2015:1440-1448.
[3] REN S,HE K,GRISHICK R,et al.Faster R-CNN:Towards real-time object detection with region proposal networks[C]//Advances in Neural Information Processing Systems.2015:91-99.
[4] GRICHICK R,DONAHUE J,DARRELL T,et al.Rich feature hierarchies for accurate object detection and semantic segmentation[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.2014:580-587.
[5] HE K M,GKIOXARI G,DOLLAR P,et al.Mask R-CNN[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2020,42(2):386-397.
[6] CHANU M M.A deep learning approach for object detectionand instance segmentation using mask RCNN[J].Journal of Advanced Research in Dynamical and Control Systems,2020,12(SP3):95-104.
[7] REDMON J,DIVVALA S,GIRSHICK R,et al.You onlylookonce:Unified,real-timeobject detection[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Re-cognition.2016:779-788.
[8] LIU W,ANGUELOV D,ERHAN D,et al.SSD:single shotmultibox detector[C]//Proceedings of the 2016 European Conference on Computer Vision.Cham:Springer,2016:21-37.
[9] REDMON J,FARHADI A.YOLO9000:better,faster,stronger[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.2017:7263-7271.
[10] REDMON J,FARHADI A.YOLOv3:an incremental impro- vement[J].arXiv:1804.02767,2018.
[11] XU Z F,JIA R S,SUN H M,et al.Light-YOLOv3:fast method for detecting green mangoes in complex scenes using picking robots[J].Applied Intelligence,2020(6):1-18.
[12] PANG L,LIU H,CHEN Y,et al.Real-time Concealed Object Detection from Passive Millimeter Wave Images Based on the YOLOv3 Algorithm[J].Sensors(Basel,Switzerland),2020,20(6):44-50.
[13] ZHAO L,LI S.Object Detection Algorithm Based on Improved YOLOv3[J].Electronics,2020,9(3):537.
[14] DENG Z,YANG R,LAN R,et al.SE-IYOLOv3:An Accurate Small Scale Face Detector for Outdoor Security[J].Mathema-tics,2020,8(1):93.
[15] DENG H,CHENG J,LIU T,et al.Research on iron surface crack detection algorithm based on improve YOLOv4 network[C]//Journal of Physics:Conference Series,2nd International Conference on Artificial Intelligence and Computer Science,Hangzhou,Zhejiang,China,2020:25-26.
[16] GU Y H,CAO M T,XIU J Y,et al.Violation Detection algorithm based on YOLOv4 Network[J].Journal of Chongqing University of Technology (Natural Science),2021,35(8):114-121.
[17] HOWARD A,SANDLER M,CHU G,et al.Searching for mobilenetv3[C]//Proceedings of the IEEE International Confe-rence on Computer Vision.2019:1314-1324.
[18] PAN M Y,SONG H H,ZHANG K H,et al.Learning Global Guided Progressive Feature Aggregation Lightweight Network for Salient Object Detection[J].Computer Science,2021,48(6):103-109.
[19] 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.
[20] BACHEM O,LUCIC M,HASSANI S H,et al.Approximate k-means++in sublinear time[C]//Proceedings of the Thirtieth AAAI Conference on Artificial Intelligence,2016:1459-1467.
[21] BACHEM O,LUCIC M,HASSANI H,et al.Fast and provably good seedings for k-means[C]//30th Conference on Neural Information Processing Systems,2016:55-63.
[22] REZATOFIGHI H,TSOI N,GWAK J Y,et al.Generalized intersection over union:a metric and a loss for bounding box regression[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.2019:658-666.
[23] REN S Q,HE K M,GIRSHICK R,et al.Faster R-CNN:to-wards real-time object detection with region proposal networks[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2017,39(6):1137-1149.
[1] LIU Dong-mei, XU Yang, WU Ze-bin, LIU Qian, SONG Bin, WEI Zhi-hui. Incremental Object Detection Method Based on Border Distance Measurement [J]. Computer Science, 2022, 49(8): 136-142.
[2] WANG Can, LIU Yong-jian, XIE Qing, MA Yan-chun. Anchor Free Object Detection Algorithm Based on Soft Label and Sample Weight Optimization [J]. Computer Science, 2022, 49(8): 157-164.
[3] ZHANG Ying-tao, ZHANG Jie, ZHANG Rui, ZHANG Wen-qiang. Photorealistic Style Transfer Guided by Global Information [J]. Computer Science, 2022, 49(7): 100-105.
[4] CHENG Cheng, JIANG Ai-lian. Real-time Semantic Segmentation Method Based on Multi-path Feature Extraction [J]. Computer Science, 2022, 49(7): 120-126.
[5] MENG Yue-bo, MU Si-rong, LIU Guang-hui, XU Sheng-jun, HAN Jiu-qiang. Person Re-identification Method Based on GoogLeNet-GMP Based on Vector Attention Mechanism [J]. Computer Science, 2022, 49(7): 142-147.
[6] YU Shu-hao, ZHOU Hui, YE Chun-yang, WANG Tai-zheng. SDFA:Study on Ship Trajectory Clustering Method Based on Multi-feature Fusion [J]. Computer Science, 2022, 49(6A): 256-260.
[7] YANG Yue, FENG Tao, LIANG Hong, YANG Yang. Image Arbitrary Style Transfer via Criss-cross Attention [J]. Computer Science, 2022, 49(6A): 345-352.
[8] CHEN Yong-ping, ZHU Jian-qing, XIE Yi, WU Han-xiao, ZENG Huan-qiang. Real-time Helmet Detection Algorithm Based on Circumcircle Radius Difference Loss [J]. Computer Science, 2022, 49(6A): 424-428.
[9] SUN Jie-qi, LI Ya-feng, ZHANG Wen-bo, LIU Peng-hui. Dual-field Feature Fusion Deep Convolutional Neural Network Based on Discrete Wavelet Transformation [J]. Computer Science, 2022, 49(6A): 434-440.
[10] LAN Ling-xiang, CHI Ming-min. Remote Sensing Change Detection Based on Feature Fusion and Attention Network [J]. Computer Science, 2022, 49(6): 193-198.
[11] CHEN Jia-zhou, ZHAO Yi-bo, XU Yang-hui, MA Ji, JIN Ling-feng, QIN Xu-jia. Small Object Detection in 3D Urban Scenes [J]. Computer Science, 2022, 49(6): 238-244.
[12] LI Fa-guang, YILIHAMU·Yaermaimaiti. Real-time Detection Model of Insulator Defect Based on Improved CenterNet [J]. Computer Science, 2022, 49(5): 84-91.
[13] DONG Qi-da, WANG Zhe, WU Song-yang. Feature Fusion Framework Combining Attention Mechanism and Geometric Information [J]. Computer Science, 2022, 49(5): 129-134.
[14] LI Peng-zu, LI Yao, Ibegbu Nnamdi JULIAN, SUN Chao, GUO Hao, CHEN Jun-jie. Construction and Classification of Brain Function Hypernetwork Based on Overlapping Group Lasso with Multi-feature Fusion [J]. Computer Science, 2022, 49(5): 206-211.
[15] HU Fu-yuan, WAN Xin-jun, SHEN Ming-fei, XU Jiang-lang, YAO Rui, TAO Zhong-ben. Survey Progress on Image Instance Segmentation Methods of Deep Convolutional Neural Network [J]. Computer Science, 2022, 49(5): 10-24.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!