損失関数は、オブジェクト検出タスクの検出精度に影響を与える重要な要素です。この論文では、分類とローカリゼーションを同時に行うために、2つの損失関数の両方を最適化します。まず、IoUベースの係数に分類損失関数の標準クロスエントロピー損失を乗算することにより、ローカリゼーションと分類の間の相関関係が確立されます。正のサンプルのローカリゼーション精度を向上させるためにのみ相関が適用される既存の研究と比較して、この論文では、相関を利用して非常に硬い負のサンプルを取得し、負のサンプルの誤分類率を減らすことを目的としています。さらに、予測ボックスとターゲットボックスの間にマハラノビス距離を組み込むことにより、MIoUという名前の新しいローカリゼーション損失が提案されます。これにより、DIoU損失の勾配の不一致の問題が解消され、ローカリゼーションの精度がさらに向上します。最後に、夜間の車両検出のための十分な実験が2つのデータセットで行われました。我々の結果は、提案された損失関数でトレーニングする場合よりも、検出性能を大幅に改善できることを示しています。ソースコードとトレーニング済みモデルは、https://github.com/therebellll/NegIoU-PosIoU-Miouで入手できます。
Loss functions is a crucial factor that affecting the detection precision in object detection task. In this paper, we optimize both two loss functions for classification and localization simultaneously. Firstly, by multiplying an IoU-based coefficient by the standard cross entropy loss in classification loss function, the correlation between localization and classification is established. Compared to the existing studies, in which the correlation is only applied to improve the localization accuracy for positive samples, this paper utilizes the correlation to obtain the really hard negative samples and aims to decrease the misclassified rate for negative samples. Besides, a novel localization loss named MIoU is proposed by incorporating a Mahalanobis distance between predicted box and target box, which eliminate the gradients inconsistency problem in the DIoU loss, further improving the localization accuracy. Finally, sufficient experiments for nighttime vehicle detection have been done on two datasets. Our results show than when train with the proposed loss functions, the detection performance can be outstandingly improved. The source code and trained models are available at https://github.com/therebellll/NegIoU-PosIoU-Miou.