A lightweight network face detection based on YOLOv5 Lightweight model face detection based on YOLOv5 combined with Mobilenetv2

被引:0
|
作者
Xu, Bowen [1 ]
Wang, Chunmei [1 ]
Yu, Baocheng [1 ]
Xu, Wenxia [1 ]
Du, Bing [2 ]
机构
[1] Wuhan Inst Technol, Engn Res Ctr Intelligent Prod Line Equipment Hube, Wuhan, Peoples R China
[2] Fiberview Technol Co Ltd, Wuhan 430074, Hubei, Peoples R China
关键词
YOLOv5; MobileNetv2; SIoU; Lightweight network; Object detection;
D O I
10.1145/3655532.3655557
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Face detection is a mainstream way of human identification today, but the current object detection network requires a large number of parameters and calculations, in response to this problem, we propose a lightweight improved face detection method based on YOLOv5, which can be easily deployed in embedded devices and mobile devices. We replaced YOLOv5s backbone with a more lightweight MobileNet, replacing standard convolution with deep separable convolution; Then we also use the SIoU loss function to speed up the convergence speed of the prediction box and the target box, improving the inference rate. Compared with the baseline model, the YOLOv5-MS model reduces the number of parameters by 91.4% and increases FPS by 62.5%, which achieves a good balance between detection accuracy and rate.
引用
收藏
页码:157 / 162
页数:6
相关论文
共 50 条
  • [1] A lightweight vehicles detection network model based on YOLOv5
    Dong, Xudong
    Yan, Shuai
    Duan, Chaoqun
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2022, 113
  • [2] An Improved Lightweight YOLOv5 Model Based on Attention Mechanism for Face Mask Detection
    Xu, Sheng
    Guo, Zhanyu
    Liu, Yuchi
    Fan, Jingwei
    Liu, Xuxu
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING - ICANN 2022, PT III, 2022, 13531 : 531 - 543
  • [3] Lightweight Road Damage Detection Network Based on YOLOv5
    Zhao, Jingwei
    Tao, Ye
    Zhang, Zhixian
    Huang, Chao
    Cui, Wenhua
    ENGINEERING LETTERS, 2024, 32 (08) : 1708 - 1720
  • [4] Fast ship detection based on lightweight YOLOv5 network
    Zheng, Jia-Chun
    Sun, Shi-Dan
    Zhao, Shi-Jia
    IET IMAGE PROCESSING, 2022, 16 (06) : 1585 - 1593
  • [5] A lightweight tea bud detection model based on Yolov5
    Gui, Zhiyong
    Chen, Jianneng
    Li, Yang
    Chen, Zhiwei
    Wu, Chuanyu
    Dong, Chunwang
    COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2023, 205
  • [6] A lightweight waxberry fruit detection model based on YOLOv5
    Yang, Chengyu
    Liu, Jun
    He, Jianting
    IET IMAGE PROCESSING, 2024, 18 (07) : 1796 - 1808
  • [7] A Lightweight Model Based on YOLOv5 for Helmet Wearing Detection
    Zou, Xiongxin
    Chen, Zuguo
    Zhou, Yimin
    4TH INTERNATIONAL CONFERENCE ON INFORMATICS ENGINEERING AND INFORMATION SCIENCE (ICIEIS2021), 2022, 12161
  • [8] A lightweight bus passenger detection model based on YOLOv5
    Li, Xiaosong
    Wu, Yanxia
    Fu, Yan
    Zhang, Lidan
    Hong, Ruize
    IET IMAGE PROCESSING, 2023, 17 (14) : 3927 - 3937
  • [9] Lightweight safflower cluster detection based on YOLOv5
    Guo, Hui
    Wu, Tianlun
    Gao, Guomin
    Qiu, Zhaoxin
    Chen, Haiyang
    SCIENTIFIC REPORTS, 2024, 14 (01):
  • [10] Lightweight network for insulator fault detection based on improved YOLOv5
    Weng, Dehua
    Zhu, Zhiliang
    Yan, Zhengbing
    Wu, Moran
    Jiang, Ziang
    Ye, Nan
    CONNECTION SCIENCE, 2024, 36 (01)