Model Zoo

November 29, 2020 ยท View on GitHub

The evaluation metrics are top-1 and top-5 video-level accuracy on Kinetics-400 validation set. The Finetuned model column provides models either trained from scratch or fine-tuned on Kinetics. The Pre-trained model column provides models pre-trained on Sports1M or IG-65M without further finetuning on Kinetics.

C3D

Input sizePre-trained datasetPre-trained modelVideo@1Video@5Finetuned modelGFLOPsparams(M)
16x112x112NoneNone66.686.7link38.564.9
16x112x112Sports1Mlink67.487.2link38.564.9

R(2+1)D-34

Input sizePre-trained datasetPre-trained modelVideo@1Video@5Finetuned modelGFLOPsparams(M)
8x112x122IG-65Mlink74.991.8link49.863.6
32x112x122IG-65Mlink80.094.2link199.063.6

R(2+1)D-152

Input sizePre-trained datasetPre-trained modelVideo@1Video@5Finetuned modelGFLOPsparams(M)
32x112x122NoneNone77.392.5link329.1118.0
32x112x122Sports1Mlink79.594.0link329.1118.0
32x112x112IG-65Mlink81.695.3link329.1118.0

ir-CSN-50

Input sizePre-trained datasetPre-trained modelVideo@1Video@5Finetuned modelGFLOPsparams(M)
32x224x224NoneNoneTBDTBDlinkTBDTBD
32x224x224IG-65MlinkTBDTBDlinkTBDTBD

ir-CSN-152

Input sizePre-trained datasetPre-trained modelVideo@1Video@5Finetuned modelGFLOPsparams(M)
32x224x224NoneNone76.592.1link96.729.6
32x224x224Sports1Mlink78.293.0link96.729.6
32x224x224IG-65Mlink82.695.3link96.729.6

ip-CSN-152

Input sizePre-trained datasetPre-trained modelVideo@1Video@5Finetuned modelGFLOPsparams(M)
32x224x224NoneNone77.892.8link108.832.8
32x224x224Sports1Mlink78.893.5link108.832.8
32x224x224IG-65Mlink82.595.3link108.832.8

Gradient-Blending: Audio-Visual Model: ip-CSN-152

Input sizePre-trained datasetPre-trained modelVideo@1Video@5Finetuned modelGFLOPs
32x224x224NoneNone79.193.9link110.1
32x224x224Sports1M+AudioSetlink80.494.8link110.1
32x224x224IG-65M+AudioSetlink83.396.0link110.1

We additionally provide two audio-visual models:

  • R(2+1)D-101 with 16x224x224 as input trained from scratch on AudioSet: link
  • ip-CSN-152 pretrained on IG-65M and ImageNet, which is used for our entry in EPIC-Kitchen Action Recognition Challenge: link