Welcome to Journal of Graphics share: 

Journal of Graphics ›› 2022, Vol. 43 ›› Issue (5): 791-802.DOI: 10.11996/JG.j.2095-302X.2022050791

• Image Processing and Computer Vision • Previous Articles     Next Articles

An object detection method of falling person based on optimized YOLOv5s 

  

  1. School of Information Engineering, Chang’an University, Xi’an Shaanxi 710064, China
  • Online:2022-10-31 Published:2022-10-28
  • Supported by:
    National Natural Science Foundation of China (51678061) 

Abstract:

To address the problems of easy missing, poor robustness and generalization ability when object detection model is detecting a person falling down, a new detection method YOLOv5s-FPD was proposed based on the improved YOLOv5s. Firstly, the Le2i fall detection data set was expanded in various ways for model training to enhance model robustness and generalization ability. Secondly, MobileNetV3 was employed as the backbone network for feature extraction, which could coordinate and balance the relationship between lightness and accuracy of the model. Furthermore, BiFPN (bi-directional feature pyramid network) was utilized to boost model multi-scale feature fusion ability, thereby improving the efficiency and speed of fusion. Meanwhile, the CBAM (convolutional block attention module) lightweight attention mechanism was adopted to realize double focus attention to channel and space, enhancing the effect of attention mechanism on model accuracy. Finally, Focal Loss evaluation was used to pay more attention to hard example mining and alleviate the samples imbalance problem. The experimental results show that the precision, F1 score, and detection speed of YOLOv5s-FPD model were improved by 2.91%, 0.03, and 8.7 FPS, respectively, compared with the original YOLOv5s model on Le2i fall detection dataset, which verified the effectiveness of the proposed method. 

Key words:  , object detection, YOLOv5s, MobileNetV3, lightweight attention, multi-scale feature fusion, focal loss function

CLC Number: