このペーパーでは、AI City Challenge 2021(AICITY21)のTrack2のソリューションを紹介します。 Track2は、実世界のデータと合成データの両方を使用する車両再識別(ReID)タスクです。この課題では、主に4つのポイント、つまりトレーニングデータ、教師なしドメイン適応(UDA)トレーニング、後処理、モデルアンサンブルに焦点を当てます。 (1)トリミングトレーニングデータと合成データの使用の両方が、モデルがより識別力のある特徴を学習するのに役立ちます。 (2)テストセットにはトレーニングセットに表示されない新しいシナリオがあるため、UDAメソッドはチャレンジでうまく機能します。 (3)再ランク付け、画像からトラックへの検索、カメラ間の融合などの後処理技術により、最終的なパフォーマンスが大幅に向上します。 (4)異なる表現の多様性を提供するCNNベースのモデルとトランスフォーマーベースのモデルをアンサンブルします。前述の手法により、私たちの方法は最終的に0.7445 mAPスコアを達成し、競合他社で1位になりました。コードはhttps://github.com/michuanhaohao/AICITY2021_Track2_DMTで入手できます。
This paper introduces our solution for the Track2 in AI City Challenge 2021 (AICITY21). The Track2 is a vehicle re-identification (ReID) task with both the real-world data and synthetic data. We mainly focus on four points, i.e. training data, unsupervised domain-adaptive (UDA) training, post-processing, model ensembling in this challenge. (1) Both cropping training data and using synthetic data can help the model learn more discriminative features. (2) Since there is a new scenario in the test set that dose not appear in the training set, UDA methods perform well in the challenge. (3) Post-processing techniques including re-ranking, image-to-track retrieval, inter-camera fusion, etc, significantly improve final performance. (4) We ensemble CNN-based models and transformer-based models which provide different representation diversity. With aforementioned techniques, our method finally achieves 0.7445 mAP score, yielding the first place in the competition. Codes are available at https://github.com/michuanhaohao/AICITY2021_Track2_DMT.