Collaborative learning in bounding box regression for object detection

Xian Fang

doi:10.1016/j.patrec.2021.05.007

2021-08-01

PRE

Xian Fang

张瑞勋 (Ruixun Zhang)

Xiuli Shao

王鸿鹏 (Hongpeng Wang) *

DOI：10.1016/j.patrec.2021.05.007

原文链接

原文求助

摘要

En 中文

Object detection has attracted growing attention due to its extensive application prospect, in which bounding box regression is an essential component. Dedicated to collaborative learning in bounding box regression, we explore the unified framework of smooth L-1 and intersection over union, named SLIoU. On the basis of that, we propose a SLIoU loss as localization loss, which focuses on the geometric relationships of pairs of rectangular bounding boxes in overlapping degree, central position and structural shape. Furthermore, we propose a SLIoU-NMS for suppressing redundant detection boxes, which adaptively maps the evaluation value of detection boxes to meet the evaluation metric using nonlinear representation. By incorporating SLIoU loss and SLIoU-NMS into the state-of-the-art one-stage detectors, the detection performance is considerably improved. (C) 2021 Elsevier B.V. All rights reserved.

Keyword:

Object detection

Bounding box regression

One-stage detector

Loss function

Non-maximum suppression