arXiv:2508.14343cs.CVcs.AI2025-08

通过物体间空间关系提升小目标检测精度,尤其适用于车牌识别。

Inter-Class Relational Loss for Small Object Detection: A Case Study on License Plates

  • 利用大物体与小物体的相对位置关系设计新损失函数
  • 在YOLOv12-T上提升mAP 10.3%,UAV-DETR提升1.6%
  • 可无缝集成到现有检测模型中,适合小目标场景应用

单阶段多目标检测中,基于交并比(IoU)的损失函数在训练接近目标时虽能实现平滑收敛,但对小目标梯度更新效果差,因梯度极平。当多个目标同时学习时,小目标梯度更新更不足。为此,本文提出一种跨类关系损失(ICR Loss),基于物体间的空间关联性(如车牌通常位于汽车上)进行优化。若预测的车牌框不在其所属汽车框内,则施加惩罚,该惩罚与车与车牌框重叠面积成反比。通过利用跨类空间关系,损失引导小目标学习,增强深层特征图中的隐含信息。以车牌检测为案例,本文贡献包括:(1) 构建首个涵盖多样真实场景、标注高质量的小型车辆多车牌数据集(SVMLP);(2) 提出新型跨类关系损失函数,显著提升检测性能。ICR损失可轻松嵌入现有IoU损失,无需额外调参即可在标准mAP指标上取得提升:YOLOv12-T提升10.3%,UAV-DETR提升1.6%。代码与数据集即将开源。

原文摘要 · Abstract (English)

In one-stage multi-object detection tasks, various intersection over union (IoU)-based solutions aim at smooth and stable convergence near the targets during training. However, IoU-based losses fail to correctly update the gradient of small objects due to an extremely flat gradient. During the update of multiple objects, the learning of small objects' gradients suffers more because of insufficient gradient updates. Therefore, we propose an inter-class relational loss to efficiently update the gradient of small objects while not sacrificing the learning efficiency of other objects based on the simple fact that an object has a spatial relationship to another object (e.g., a car plate is attached to a car in a similar position). When the predicted car plate's bounding box is not within its car, a loss punishment is added to guide the learning, which is inversely proportional to the overlapped area of the car's and predicted car plate's bounding box. By leveraging the spatial relationship at the inter-class level, the loss guides small object predictions using larger objects and enhances latent information in deeper feature maps. In this paper, we present twofold contributions using license plate detection as a case study: (1) a new small vehicle multi-license plate dataset (SVMLP), featuring diverse real-world scenarios with high-quality annotations; and (2) a novel inter-class relational loss function designed to promote effective detection performance. We highlight the proposed ICR loss penalty can be easily added to existing IoU-based losses and enhance the performance. These contributions improve the standard mean Average Precision (mAP) metric, achieving gains of 10.3% and 1.6% in mAP$^{\text{test}}_{50}$ for YOLOv12-T and UAV-DETR, respectively, without any additional hyperparameter tuning. Code and dataset will be available soon.

小目标检测跨类关系车牌识别损失函数

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。