Chinese Journal of Liquid Crystals and Displays, Volume. 38, Issue 5, 656(2023)

Object detection method based on CIoU improved bounding box loss function

Xiong-biao LIU1,2, Xian-zhao YANG1,2、*, Yang CHEN1,2, and Shuai-tong ZHAO1,2
Author Affiliations
  • 1College of Information Science and Engineering,Wuhan University of Science and Technology,Wuhan 430081,China
  • 2Engineering Research Center of Metallurgical Automation and Measurement Technology,Ministry of Education,Wuhan 430081,China
  • show less

    The loss function plays an important role in the detection accuracy and model convergence speed of the object detection task, and the bounding box loss function in the loss function is an important factor affecting the detection results and model convergence speed. To address the problems of low accuracy for traditional model localization and slow convergence of the model during training, an improved bounding box loss function is proposed based on the CIoU bounding box loss function, which solves the problem of gradient explosion and early degradation of the model brought by the bounding box aspect ratio during the derivation of CIoU loss function, and introduces the normalized distance between the aspect relationship and centroid of the overlap region and the target box as additional penalty terms to improve the detection accuracy and convergence speed of the model, which is called BCIoU (Better CIoU). Experimental results on the PASACL VOC 2007 dataset show that the improved BCIoU bounding box loss function improves the IoU loss mAP50 metric by 2.09% and the AP metric by 6.88% relatively under the YOLOv3 network, the CIoU loss mAP50 metric improves by 1.64% and the AP metric improves by 5.35%. The convergence speed of the model is also improved to some extent. The proposed BCIoU loss function improves detection accuracy and convergence speed of the model, and can be easily incorporated into current object detection algorithms.

    Tools

    Get Citation

    Copy Citation Text

    Xiong-biao LIU, Xian-zhao YANG, Yang CHEN, Shuai-tong ZHAO. Object detection method based on CIoU improved bounding box loss function[J]. Chinese Journal of Liquid Crystals and Displays, 2023, 38(5): 656

    Download Citation

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

    Category: Research Articles

    Received: Aug. 25, 2022

    Accepted: --

    Published Online: Jul. 4, 2023

    The Author Email: Xian-zhao YANG (yangxianzhao@wust.edu.cn)

    DOI:10.37188/CJLCD.2022-0282

    Topics