一种角度自适应的椭圆模板检测器
An angle - adaptive elliptic template detector
投稿时间: 2022/12/20 0:00:00
DOI:
中文关键词: 目标检测;旋转椭圆框;损失函数
英文关键词: object detection; oriented elliptical box; loss function
基金项目: 国家自然科学基金(U19B2039, 61906011, 61632006, 61772048, 61672071, U1811463, 61806014)
姓名 单位
胡永利 北京工业大学人工智能与自动化学院
武剑 北京工业大学人工智能与自动化学院
孙艳丰 北京工业大学人工智能与自动化学院
点击数:589 下载数:455
中文摘要:

本文提出了一种新颖的摆脱锚框的椭圆模板检测器EllipseNet,其能够直接预测椭圆框的圆心,长短轴长以及旋转角度。由于椭圆模板检测网络能够自适应的调整旋转角度故能够更加精确的检测出物体的轮廓。另外,为了提高模型的检测精度,设计了一种高效的椭圆交并比损失函数。实验结果表明,由于没有引入复杂的锚框,该模型结构简单,且取得了和基于矩形框目标检测算法相近的性能。

英文摘要:

This paper propose a novel anchor-free object detector, namely EllipseNet. Instead of using the rectangular box, EllipseNet detects an elliptical box by predicting the center, axis and tilt angle of the ellipse. As the ellipse model can capture the object rotation and match the target adaptively by adjusting tilt angle, EllipseNet has more accurate detection boundary. In addition, we introduce an Elliptical IOU Loss to improve the detection performance of the model. The experimental results show that the model has simple structure and achieves similar performance to the object detection methods based on rectangular boxes because there is no complex anchor boxes introduced.

参考文献: