Chinese Journal of Liquid Crystals and Displays, Volume. 38, Issue 7, 945(2023)

Lightweight and high-precision object detection algorithm based on YOLO framework

Xin-chuan FAN and Chun-mei CHEN*
Author Affiliations
  • School of Information Engineering,Southwest University of Science and Technology,Mianyang 621010,China
  • show less

    Image-oriented multi-scale object detection algorithms often have the problem of mutual restriction between detection accuracy and system cost. Therefore, a lightweight and high-precision object detection algorithm based on YOLO framework is proposed. Under the YOLO framework, the mechanism of down-sampling and channel attention based on MobileNetv3 network is improved to accurately extract target features and reduce unnecessary overhead. The feature pyramid and single-stage headless fusion structure are designed, and different receptive fields are constructed to obtain different scale information, so as to enhance the adaptability of the algorithm for multi-scale targets. At the same time, SIOU is used as regression loss and Soft-NMS is used for redundant frame processing to improve the accuracy of the algorithm. Experiments are conducted on the MS COCO and UA-DETRAC. Compared with the original YOLOXs, the results show that the proposed improved algorithm reduces the number of model parameters and the computational cost reduced by 64.98% and 57.14% without reducing the accuracy. On the UA-DETRAC, mAP@0.5 reaches 70.5% which is improved by 3.52%, and FPS increases by 14.4%. The experimental results show that our algorithm greatly reduces the system overhead, improves the accuracy, and effectively guarantees the dual performance of detection.

    Tools

    Get Citation

    Copy Citation Text

    Xin-chuan FAN, Chun-mei CHEN. Lightweight and high-precision object detection algorithm based on YOLO framework[J]. Chinese Journal of Liquid Crystals and Displays, 2023, 38(7): 945

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Research Articles

    Received: Nov. 13, 2022

    Accepted: --

    Published Online: Jul. 31, 2023

    The Author Email: Chun-mei CHEN (47920787@qq.com)

    DOI:10.37188/CJLCD.2022-0328

    Topics