File size: 465 Bytes
3bbb319 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# Copyright (c) OpenMMLab. All rights reserved.
import torch
from mmpose.models.backbones import I3D
def test_i3d_backbone():
"""Test I3D backbone."""
model = I3D()
model.train()
vids = torch.randn(1, 3, 16, 112, 112)
feat = model(vids)
assert feat.shape == (1, 1024, 2, 3, 3)
model = I3D(expansion=0.5)
model.train()
vids = torch.randn(1, 3, 32, 224, 224)
feat = model(vids)
assert feat.shape == (1, 512, 4, 7, 7)
|