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 条
  • [31] Lightweight Surface Defect Detection Algorithm Based on Improved YOLOv5
    Yang, Kaijun
    Chen, Tao
    2024 5TH INTERNATIONAL CONFERENCE ON MECHATRONICS TECHNOLOGY AND INTELLIGENT MANUFACTURING, ICMTIM 2024, 2024, : 798 - 802
  • [32] Research on lightweight algorithm for gangue detection based on improved Yolov5
    Yuan, Xinpeng
    Fu, Zhibo
    Zhang, Bowen
    Xie, Zhengkun
    Gan, Rui
    SCIENTIFIC REPORTS, 2024, 14 (01)
  • [33] Research on lightweight algorithm for gangue detection based on improved Yolov5
    Xinpeng Yuan
    Zhibo Fu
    Bowen Zhang
    Zhengkun Xie
    Rui Gan
    Scientific Reports, 14
  • [34] A Lightweight Military Target Detection Algorithm Based on Improved YOLOv5
    Du, Xiuli
    Song, Linkai
    Lv, Yana
    Qiu, Shaoming
    ELECTRONICS, 2022, 11 (20)
  • [35] Lightweight tea bud detection method based on improved YOLOv5
    Zhang, Kun
    Yuan, Bohan
    Cui, Jingying
    Liu, Yuyang
    Zhao, Long
    Zhao, Hua
    Chen, Shuangchen
    SCIENTIFIC REPORTS, 2024, 14 (01):
  • [36] Lightweight detection algorithm of seed potato eyes based on YOLOv5
    Gu, Hongyu
    Li, Zhihe
    Li, Tao
    Li, Tianhao
    Li, Ning
    Wei, Zhongcai
    Nongye Gongcheng Xuebao/Transactions of the Chinese Society of Agricultural Engineering, 2024, 40 (14): : 126 - 136
  • [37] An improved lightweight object detection algorithm for YOLOv5
    Luo, Hao
    Wei, Jiangshu
    Wang, Yuchao
    Chen, Jinrong
    Li, Wujie
    PEERJ COMPUTER SCIENCE, 2024, 10
  • [38] YOLOv5-R: lightweight real-time detection based on improved YOLOv5
    Ren, Jian
    Wang, Zhijie
    Zhang, Yifan
    Liao, Lei
    JOURNAL OF ELECTRONIC IMAGING, 2022, 31 (03)
  • [39] An Improved Face Mask Detection Simulation Algorithm Based on YOLOv5 Model
    Qi, Yue
    Wang, Yiqin
    Dong, Yunyun
    INTERNATIONAL JOURNAL OF GAMING AND COMPUTER-MEDIATED SIMULATIONS, 2024, 16 (01)
  • [40] Improved YOLOv5 Lightweight Mask Detection Algorithm
    Liu, Chonghao
    Pan, Lihu
    Yang, Fan
    Zhang, Rui
    Computer Engineering and Applications, 2023, 59 (07) : 232 - 241