Model Zoo

May 5, 2022 · View on GitHub

StageModelARID Top-1DownloadShell
Pre-trainUniFormer-B32-K600N/AgoogleN/A
Pre-trainUniFormer-B32-SSV2N/AgoogleN/A
Pre-trainMViT-B32-K600N/AgoogleN/A
Pre-trainSlowFast-R101-K700N/AgoogleN/A
Adapt BNUniFormer-B32-K60062.64googlerun.sh
Adapt BNUniFormer-B32-SSV258.90googlerun.sh
Adapt BNMViT-B32-K60058.14googlerun.sh
Adapt BNSlowFast-R101-K70057.79googlerun.sh
Pseudo1UniFormer-B32-K60083.50googlerun.sh
Pseudo1UniFormer-B32-SSV281.04googlerun.sh
Pseudo1MViT-B32-K60081.68googlerun.sh
Pseudo1SlowFast-R101-K70080.78googlerun.sh
Pseudo2UniFormer-B32-K60087.84googlerun.sh
Pseudo2UniFormer-B32-SSV285.95googlerun.sh
Pseudo2MViT-B32-K60086.63googlerun.sh
Pseudo2SlowFast-R101-K70085.63googlerun.sh
Pseudo3UniFormer-B32-K60089.48googlerun.sh
Pseudo3UniFormer-B32-SSV288.74googlerun.sh
Pseudo3MViT-B32-K60088.75googlerun.sh
Pseudo3SlowFast-R101-K70088.59googlerun.sh
Pseudo4UniFormer-B32-K60089.91googlerun.sh
Pseudo4UniFormer-B32-SSV290.25googlerun.sh
Pseudo4MViT-B32-K60090.30googlerun.sh
Pseudo4SlowFast-R101-K70089.49googlerun.sh
Pseudo4UniFormer-B32-SSV2†89.51googlerun.sh

Note:

  1. All models are trained with 32 frames that are uniformly sampled from the raw videos by default, except that the UniFormer-B32† is trained with dense sampling.
  2. We used all the videos in ARID (a total of 6207 videos) for validation. For training, we generate pseudo labels for these videos.
  3. All acc results are evaluated with the TTA of 1(crop)x1(view) and gamma correction.

You can reuse all these models via setting TRAIN.CHECKPOINT_FILE_PATH and TEST.CHECKPOINT_FILE_PATH.