Shape-IoU: More Accurate Metric considering Bounding Box Shape and Scale

• URL: http://arxiv.org/abs/2312.17663v2
• Date: Fri, 12 Jan 2024 15:28:22 GMT
• Title: Shape-IoU: More Accurate Metric considering Bounding Box Shape and Scale
• Authors: Hao Zhang, Shuaijie Zhang
• Abstract summary: The Shape IoU method can calculate the loss by focusing on the shape and scale of the bounding box itself. Our method can effectively improve detection performance and outperform existing methods, achieving state-of-the-art performance in different detection tasks.
• Score: 5.8666339171606445
• License: http://creativecommons.org/licenses/by/4.0/
• Abstract: As an important component of the detector localization branch, bounding box regression loss plays a significant role in object detection tasks. The existing bounding box regression methods usually consider the geometric relationship between the GT box and the predicted box, and calculate the loss by using the relative position and shape of the bounding boxes, while ignoring the influence of inherent properties such as the shape and scale of the bounding boxes on bounding box regression. In order to make up for the shortcomings of existing research, this article proposes a bounding box regression method that focuses on the shape and scale of the bounding box itself. Firstly, we analyzed the regression characteristics of the bounding boxes and found that the shape and scale factors of the bounding boxes themselves will have an impact on the regression results. Based on the above conclusions, we propose the Shape IoU method, which can calculate the loss by focusing on the shape and scale of the bounding box itself, thereby making the bounding box regression more accurate. Finally, we validated our method through a large number of comparative experiments, which showed that our method can effectively improve detection performance and outperform existing methods, achieving state-of-the-art performance in different detection tasks.Code is available at https://github.com/malagoutou/Shape-IoU

Related papers

• Theoretically Achieving Continuous Representation of Oriented Bounding Boxes [64.15627958879053]
  This paper endeavors to completely solve the issue of discontinuity in Oriented Bounding Box representation. We propose a novel representation method called Continuous OBB (COBB) which can be readily integrated into existing detectors. For fairness and transparency of experiments, we have developed a modularized benchmark based on the open-source deep learning framework Jittor's detection toolbox JDet for OOD evaluation.
  arXiv  Detail & Related papers  (2024-02-29T09:27:40Z)
• Focaler-IoU: More Focused Intersection over Union Loss [5.8666339171606445]
  Bounding box regression plays a crucial role in the field of object detection. We analyzed the impact of difficult and easy sample distribution on regression results. We proposed Focaler-IoU, which can improve detector performance in different detection tasks.
  arXiv  Detail & Related papers  (2024-01-19T07:01:07Z)
• MPDIoU: A Loss for Efficient and Accurate Bounding Box Regression [0.0]
  We propose a novel bounding box similarity comparison metric MPDIoU. The MPDIoU loss function is applied to state-of-the-art instance segmentation (e.g., YOLACT) and object detection (e.g., YOLOv7) model trained on PASCAL VOC, MS COCO, and IIIT5k.
  arXiv  Detail & Related papers  (2023-07-14T23:54:49Z)
• Rigidity-Aware Detection for 6D Object Pose Estimation [60.88857851869196]
  Most recent 6D object pose estimation methods first use object detection to obtain 2D bounding boxes before actually regressing the pose. We propose a rigidity-aware detection method exploiting the fact that, in 6D pose estimation, the target objects are rigid. Key to the success of our approach is a visibility map, which we propose to build using a minimum barrier distance between every pixel in the bounding box and the box boundary.
  arXiv  Detail & Related papers  (2023-03-22T09:02:54Z)
• Weakly Supervised Image Segmentation Beyond Tight Bounding Box Annotations [5.000514512377416]
  This study investigates whether it is possible to maintain good segmentation performance when loose bounding boxes are used as supervision. The proposed polar transformation based MIL formulation works for both tight and loose bounding boxes. The results demonstrate that the proposed approach achieves state-of-the-art performance for bounding boxes at all precision levels.
  arXiv  Detail & Related papers  (2023-01-28T02:11:36Z)
• BoxPolyp:Boost Generalized Polyp Segmentation Using Extra Coarse Bounding Box Annotations [79.17754846553866]
  We propose a boosted BoxPolyp model to make full use of both accurate mask and extra coarse box annotations. In practice, box annotations are applied to alleviate the over-fitting issue of previous polyp segmentation models. Our proposed model outperforms previous state-of-the-art methods by a large margin.
  arXiv  Detail & Related papers  (2022-12-07T07:45:50Z)
• SIoU Loss: More Powerful Learning for Bounding Box Regression [0.0]
  Loss function SIoU was suggested, where penalty metrics were redefined considering the angle of the vector between the desired regression. Applied to conventional Neural Networks and datasets it is shown that SIoU improves both the speed of training and the accuracy of the inference.
  arXiv  Detail & Related papers  (2022-05-25T12:46:21Z)
• Object-Guided Instance Segmentation With Auxiliary Feature Refinement for Biological Images [58.914034295184685]
  Instance segmentation is of great importance for many biological applications, such as study of neural cell interactions, plant phenotyping, and quantitatively measuring how cells react to drug treatment. Box-based instance segmentation methods capture objects via bounding boxes and then perform individual segmentation within each bounding box region. Our method first detects the center points of the objects, from which the bounding box parameters are then predicted. The segmentation branch reuses the object features as guidance to separate target object from the neighboring ones within the same bounding box region.
  arXiv  Detail & Related papers  (2021-06-14T04:35:36Z)
• Rethinking Rotated Object Detection with Gaussian Wasserstein Distance Loss [111.8807588392563]
  Boundary discontinuity and its inconsistency to the final detection metric have been the bottleneck for rotating detection regression loss design. We propose a novel regression loss based on Gaussian Wasserstein distance as a fundamental approach to solve the problem.
  arXiv  Detail & Related papers  (2021-01-28T12:04:35Z)
• Dive Deeper Into Box for Object Detection [49.923586776690115]
  We propose a box reorganization method(DDBNet), which can dive deeper into the box for more accurate localization. Experimental results show that our method is effective which leads to state-of-the-art performance for object detection.
  arXiv  Detail & Related papers  (2020-07-15T07:49:05Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.