Added video inference examples and fixed a bug loading pretrained vid…

…eo models.
facebookresearch · Jun 9, 2023 · f3087b7 · f3087b7
1 parent 34142c9
commit f3087b7
Show file tree

Hide file tree

Showing 4 changed files with 189 additions and 17 deletions.
diff --git a/examples/inference.ipynb b/examples/inference.ipynb
diff --git a/examples/vid/dog.mp4 b/examples/vid/dog.mp4
diff --git a/examples/vid/goat.mp4 b/examples/vid/goat.mp4
diff --git a/hiera/hiera.py b/hiera/hiera.py
@@ -473,9 +473,9 @@ def hiera_huge_224(**kwdargs):
 @pretrained_model({
     "mae_k400_ft_k400": "https://dl.fbaipublicfiles.com/hiera/hiera_base_16x224.pth",
 }, default="mae_k400_ft_k400")
-def hiera_base_16x224(**kwdargs):
+def hiera_base_16x224(num_classes: int = 400, **kwdargs):
     return Hiera(
-        num_classes=400,  # K400 has 400 classes
+        num_classes=num_classes,  # K400 has 400 classes
         input_size=(16, 224, 224),
         q_stride=(1, 2, 2),
         mask_unit_size=(1, 8, 8),