arXiv reaDer
LABNet:車両再識別のためのクラスバランス損失を備えたローカルグラフ集約ネットワーク
LABNet: Local Graph Aggregation Network with Class Balanced Loss for Vehicle Re-Identification
車両の再識別は、さまざまな視点で見た一連の車両の中から特定の車両を識別することを目的とする重要なコンピュータビジョンタスクです。ディープラーニングに基づく最近の方法では、バックボーン特徴抽出器の後にグローバル平均プーリングレイヤーを利用しますが、これは特徴マップ上の空間的推論を無視します。本論文では、局所情報の関連性を学習し、それにより特徴学習を改善し、部分的オクルージョンと背景クラッタの影響を低減するために、バックボーン特徴マップ上でローカルグラフ集約を提案します。私たちのローカルグラフ集約ネットワークは、特徴マップの空間領域をノードと見なし、グローバル平均プーリングレイヤーの前にローカル特徴集約を実行するローカル近傍グラフを構築します。さらに、バッチ正規化レイヤーを利用して、システムの有効性を向上させます。さらに、最も広く使用されている車両再識別データセットに見られるサンプル分布の不均衡を補うために、クラスバランス損失を導入します。最後に、3つの一般的なベンチマークでメソッドを評価し、アプローチが多くの最先端のメソッドよりも優れていることを示します。
Vehicle re-identification is an important computer vision task where the objective is to identify a specific vehicle among a set of vehicles seen at various viewpoints. Recent methods based on deep learning utilize a global average pooling layer after the backbone feature extractor, however, this ignores any spatial reasoning on the feature map. In this paper, we propose local graph aggregation on the backbone feature map, to learn associations of local information and hence improve feature learning as well as reduce the effects of partial occlusion and background clutter. Our local graph aggregation network considers spatial regions of the feature map as nodes and builds a local neighborhood graph that performs local feature aggregation before the global average pooling layer. We further utilize a batch normalization layer to improve the system effectiveness. Additionally, we introduce a class balanced loss to compensate for the imbalance in the sample distributions found in the most widely used vehicle re-identification datasets. Finally, we evaluate our method in three popular benchmarks and show that our approach outperforms many state-of-the-art methods.
updated: Sun Nov 29 2020 18:43:30 GMT+0000 (UTC)
published: Sun Nov 29 2020 18:43:30 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト