Chinese Journal of Liquid Crystals and Displays, Volume. 38, Issue 11, 1468(2023)

Object detection algorithm based on adaptive focal CRIoU loss

Zhen-jiu XIAO1, Hao-ze ZHAO2, Li-li ZHANG2, Yu XIA3, Jie-long GUO4、*, Hui YU4, Cheng-long LI2, and Li-wen WANG2
Author Affiliations
  • 1College of Software Engineering,Liaoning Technical University,Huludao 125000,China
  • 2Air Ammunition Research Institute Co. Ltd.,NORINCO Group,Haerbin150000,China
  • 3Shanghai Institute of Aerospace System Engineer,Shanghai 201100,China
  • 4Quanzhou Institute of Equipment Manufacturing,Haixi Institutes,Chinese Academy of Sciences,Quanzhou 362000,China
  • show less

    In the object detection task, there is no correlation between the regression content of the traditional bounding box regression loss function and the evaluation standard IoU(Intersection over Union), and there is some irrationality for the regression attribute of the bounding box, which reduces the detection accuracy and convergence speed. In addition, the sample imbalance also exists in the regression task, and a large number of low-quality samples affect the loss function convergence. In this paper, a novel loss function, termed as CRIoU(Complete Relativity Intersection over Union),is proposed to improve the detection accuracy and convergence speed. Firstly, this work determines the design idea and determines the improved IoU loss function normal form. Secondly, on the basis of IoU loss, the ratio of the perimeter of the rectangle formed by the two center points and the minimum closure area formed by the two frames is introduced as the penalty term for the distance between the center points of the bounding box, and the improved IoU loss is applied to the non-maximum suppression.Then, the side error of the two frames and the side square of the minimum bounding box are introduced as the side penalty term,a novel loss function, termed as CRIoU(Complete Relativity Intersection over Union), is proposed. Finally, on the basis of CRIoU, an adaptive weighting factor is added to weight the regression loss of high-quality samples, and an AF-CRIoU (Adaptive focal CRIoU) is defined. The experimental results show that the detection accuracy of the AF-CRIoU loss function compared with the traditional non IoU series loss is up to 8.52%, the detection accuracy of the CIoU series loss is up to 2.69%, and the A-CRIoU NMS (Around CRIoU Non Maximum Suppression) method compared with the original NMS method is up to 0.14%. In addition, AF-CRIoU loss is applied to the detection of safety helmet, which also achieves good detection results.

    Tools

    Get Citation

    Copy Citation Text

    Zhen-jiu XIAO, Hao-ze ZHAO, Li-li ZHANG, Yu XIA, Jie-long GUO, Hui YU, Cheng-long LI, Li-wen WANG. Object detection algorithm based on adaptive focal CRIoU loss[J]. Chinese Journal of Liquid Crystals and Displays, 2023, 38(11): 1468

    Download Citation

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

    Category: Research Articles

    Received: Jan. 6, 2023

    Accepted: --

    Published Online: Nov. 29, 2023

    The Author Email: Jie-long GUO (gjl@fjirsm.ac.cn)

    DOI:10.37188/CJLCD.2023-0005

    Topics