$30 off During Our Annual Pro Sale. View Details »

社内CV輪講資料 / Kaggle NFL2

社内CV輪講資料 / Kaggle NFL2

社内のCV輪講について参加したkaggleコンペの振り返りをしたときの資料です

NFL Health & Safety - Helmet Assignment:
https://www.kaggle.com/c/nfl-health-and-safety-helmet-assignment

Takumi Karasawa

November 19, 2021
Tweet

More Decks by Takumi Karasawa

Other Decks in Programming

Transcript

  1. 5 “NFL Health & Safety - Helmet Assignment” アメフトの試合動画中のヘルメットを検出し、選⼿ラベルを割り当て NFL2

    コンペ 楽しそう。。 賞⾦⾼い。。 https://www.kaggle.com/c/nfl-health-and-safety-helmet-assignment/overview/description GIF
  2. 1プレイ(1動画) 約5~10sec 動画︓ n 59.94fps n 2つの固定視点 センサデータ︓ n 10Hz

    n 選⼿ラベルや 位置情報など ➤ボールスナップ(≒プレイ開始)が10フレーム⽬ 7 試合中の1プレイの動画とそれに対応するセンサ情報から、 動画中のヘルメットを検出し、選⼿ラベルを割り当てる タスク left, width, top, height, label
  3. 11 主データ︓ ▪ train︓120動画=60プレイ x 2視点 ▪ フレーム数︓300~600程度 ▪ test︓30動画/15プレイ(public23%/private77%)

    補助データ︓物体検出⽤の静⽌画データセット ▪ 9947(2フレーム主データと被り) データセット 各動画フレーム数 partial sideline
  4. 12 ▪ Weighted Accuracy ▪ 衝突の瞬間のヘルメット重みを1000倍 ▪ 衝突の予測は必要なし ▪ 予測ルール︓

    (各フレームにつき) ▪ 22ボックスまで ▪ 選⼿の重複ダメ ▪ ボックスの重複ダメ 評価
  5. 1プレイ(1動画) 約5~10sec 動画︓ n 59.94fps n 2つの固定視点 センサデータ︓ n 10Hz

    n 選⼿ラベルや 位置情報など ➤ボールスナップ(≒プレイ開始)が10フレーム⽬ 13 試合中の1プレイの動画とそれに対応するセンサ情報から、 動画中のヘルメットを検出し、選⼿ラベルを割り当てる タスク left, width, top, height, label 再掲 まずそもそも何する︖が少し多様な感じのコンペだった
  6. 15 全体 Helmet Detection Helmet Clustering Helmet Tracking By-frame Assignment

    Tracklet Assignment Self-Verification & Team Assignment Post Processing Jersey Recognition
  7. 16 YOLOv5(YOLOv5x6) NFL1stコンペの上位解法でよさげとされてたため ▪ 使⽤データ︓ ▪ train: 補助データセット全て ▪ val:

    主データセット (30フレーム間隔で1700枚サンプリング) ▪ w/o sidelineプレイヤー ▪ 画像特徴から区別してほしかった ▪ フル解像度1280 ▪ mAP 0.963 1. Helmet Detection Helmet Detection mp4
  8. 17 シンプルなIoUベース (IoU threshold: 0.2, max-age: 1) モチベーション︓ ▪ ⼗分に⾼fpsだった(60fps)

    ▪ ヘルメットの画像特徴、相性良くはなさそう ▪ クイックな変化へカルマンフィルタ特に嬉しくなさそう(特に衝突) ▪ IoUで明らかにわかるやつ逃したり、スイッチが散⾒ 結果的にDeepSORT, SORTをチューニングしたもの に⽐べ⾼速かつやや⾼精度だった ▪ ほか、ByteTrack-likeに低スコア拾うのを試すも特に改善なし 2. Helmet Tracking 参考)前フレームとのIoU(GT) チームレベルの活⽤はしたかった (戻ってこれず…) Helmet Detection Helmet Tracking
  9. 18 シンプルなIoUベース (IoU threshold: 0.2, max-age: 1) モチベーション︓ ▪ ⼗分に⾼fpsだった(60fps)

    ▪ ヘルメットの画像特徴、相性良くはなさそう ▪ クイックな変化へカルマンフィルタ特に嬉しくなさそう(特に衝突) ▪ IoUで明らかにわかるやつ逃したり、スイッチが散⾒ 結果的にDeepSORT, SORTをチューニングしたもの に⽐べ⾼速かつやや⾼精度だった ▪ ほか、ByteTrack-likeに低スコア拾うのを試すも特に改善なし 2. Helmet Tracking 前フレームとのIoU(GT) チームレベルの活⽤はしたかった (戻ってこれず…) Helmet Detection Helmet Tracking mp4
  10. 19 2タイプでアンサンブル(コスト⼩さい⽅選択)︓ 1. 座標変換+探索+normalizeして線形割当 (コスト最⼩を選択) 2. Homography予測➤直で線形割当 ▪ ⽐較のためコストのみnormalize 解決を図る課題感はこのへん︓

    ▪ フレーム内に存在する選⼿の特定 ▪ 画像内座標 ↔ センサ座標 ▪ 選⼿姿勢によるヘルメット位置座標の歪み 3. By-frame Assignment By-frame Assignment 座標変換+探索+線形割当 Homography予測+線形割当 Helmet Detection *fpsの違いについては最近傍のセンサ情報使⽤
  11. 20 座標変換+探索+normalizeして線形割当(コスト最⼩を選択) ➤ボールスナップ時に焦点 座標変換(sideline only)︓ ▪ カメラ位置を仮定し、 センサデータx, y座標中⼼から傾き推定 ▪

    pitchからBEV⽅向に変換したのち回転 ▪ 中⼼から遠いものは3通り探索 ▪ 直線検出なども⼀応実施 探索+normalizeして線形割当︓ ▪ センサデータx, y座標外側の選⼿から排除 ▪ 両座標を0~1にnormalizeして、 cost matrix計算後、線形割当 ▪ endzoneのみ画像y⽅向の姿勢ゆらぎ が⼤きく、x⽅向の重みを2倍 3-1. By-frame Assignment cam 3位discussionの図より https://www.kaggle.com/c/nfl-health-and-safety-helmet-assignment/discussion/285076
  12. 22 Homography予測➤直で線形割当 ▪ transformerベース ▪ ⼊⼒︓検出結果とセンサデータ ▪ アンサンブルのためコストのみnormalize 全体的なlimitation︓ ▪

    いずれにせよ座標変換ベースなので 選⼿姿勢による座標の不整合… ➤(相対𝑥!"# , 相対𝑦!"# )予測モデル検討も時間なく… 3-2. By-frame Assignment shiftも
  13. 23 基本的にtrackingへ信頼を寄せる⽅針︓ (=false splitはあれどfalse concatが少ないの意。) ▪ 同⼀tracklet内ヘルメットは同⼀ラベルでアサイン ▪ スコアリング︓バイフレームでアサインする際のコスト逆数の総和 ▪

    特にアサインロジックは動画後半間違ってるので単なるvotingだときつい ▪ フレーム内コンフリクト解消も同スコアで⼩さい⽅をドロップ 4. Tracklet Assignment V11 (0.02) V11 (0.01) V11 (0.05) None ↓ V11 V18 (0.3) ↓ V11 V11 (0.06) instance-id: 4 Helmet Detection Helmet Tracking By-frame Assignment Tracklet Assignment
  14. 24 ▪ k-meansで2クラスタ ▪ 20x20にresizeして、10x10でcenter crop ▪ ヘルメットの⾊が同じこともあるため ヘルメットのみだとlimitationあり。 ➤洗練の時間なくself-verification導⼊

    5. Helmet Clustering & Team Assignment Helmet Clustering By-frame Assignment Tracklet Assignment Self-Verification & Team Assignment ヘルメットの⾊が同じ
  15. 25 ▪ k-meansで2クラスタ ▪ 20x20にresizeして、10x10でcenter crop ▪ limitationに対して、self-verification導⼊ ▪ Self-verification︓

    ▪ 条件︓ 1. (予測結果との相関⼤)or(RGB分布が⼗分に分離) ▪ 予測性能に依存するのでRGB分布によって拾う 2. 2つのクラスタサイズが同程度 ▪ 「片方チームの一部と残り」ケースの回避(ヘルメット向きによるアピアランス変化など) ▪ 25/120動画が不適⽤となるくらい(厳しめに倒してる) ▪ Team Assignment︓ ▪ 予測結果からvotingでチーム決め ▪ 適⽤︓ ▪ アサインにて線形割当のcost matrixにて異チームの組み合わせのコスト5倍 5. Helmet Clustering & Team Assignment Helmet Clustering By-frame Assignment Tracklet Assignment Self-Verification & Team Assignment
  16. 26 easyocrを⽤いて背番号認識 ▪ recall低いがprecision⾼くて適⽤しやすかった ▪ 動画後半の拡⼤時に効く ▪ 2倍にupsamplingして⼊⼒(recall向上) 適⽤︓ ▪

    OCRしてサイズや番号でフィルタリング後、 画像内ヘルメット平均サイズ1個分上にシフトし、最近傍に割当 ▪ 有⼒候補として登録し、trackletのアサイン時に有⼒候補あればアサイン 6. Jersey Recognition (OCR) Tracklet Assignment Jersey Recognition by-frame assignmentへも 適⽤したが数分間に合わず occlusionミスが稀に (2桁➤1桁) recall⾼めの例
  17. 27 ▪ 評価指標が(submit的には)weighted recall ➤22選⼿以下の制限の下、なるべく予測を⼊れる 適⽤︓ 1. 最終予測結果とセンサデータの対応関係からhomography算出 ▪ 対応点不⾜時スキップ

    2. 割当のない検出結果をセンサ座標に投影 ▪ tracklet assignmentでのdropも多いので 3. 割当のない選⼿へ近い順にアサイン ▪ (チーム分類適⽤時)異チーム組み合わせはコスト5倍 7. Post-processing Tracklet Assignment Self-Verification & Team Assignment Post Processing
  18. 29 全体再掲 Helmet Detection Helmet Clustering Helmet Tracking By-frame Assignment

    Tracklet Assignment Self-Verification & Team Assignment Post Processing Jersey Recognition
  19. 30 各段階のスコア private public date 10/14 10/15 10/25 10/31 11/2

    11/2 11/3 11/3 約10分前完了.. ↑⼿元のスコア(privateとほぼ同じくらい) 0.665 0.792 0.806 0.821 37th 23rd 20th 20th 18th 16th 52min 53min 58min 83min 91min ↑est. inference time
  20. 33 ▪ 物体検出︓YOLOv5 ▪ トラッキング︓DeepSORT ▪ アサイン︓外側の選⼿を順に弾いて、 ICP(iterative closest point)+

    線形割当 ▪ ICP:三次元点群同⼠を位置合わせする⼿法 ▪ チーム分類︓k-means clustering ▪ DeepSORTでの特徴を使⽤ ▪ 最後に全情報を⽤いて、ICP+線形割当 3rd : fantastic_hirarinさん [link] ICP+線形割当
  21. 34 ▪ 物体検出: YOLOv5 ▪ トラッキング: SORT ▪ チーム分類: 2-phase

    k-means clustering ▪ 特徴抽出(マッチングの距離に使⽤)︓ ▪ player orientation prediction ▪ apparent gap prediction ▪ 座標変換︓直線検出 ▪ アサイン: 外側の選⼿を順に弾いて線形割当 2nd : titoさん [link] 直線検出 apparent gap チーム分類
  22. 35 ▪ 物体検出︓FCOS-likeモデル 2-stage検出 ▪ Image to Map Converter ▪

    カメラ⽅向からの2D座標を予測 ▪ アサイン︓ICPベース ▪ チーム分類︓ArcFaceで同チームかを予測 ▪ トラッキング︓IoUベース 1st : k_matさん [link] Image2Map *discussionとてもわかりやすいので詳細はそちら⾒ていただくのが良いかと︕