File size: 667 Bytes
d670799 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
# Copyright (c) OpenMMLab. All rights reserved.
import torch
from mmaction.models import C2D
from mmaction.testing import generate_backbone_demo_inputs
def test_c2d_backbone():
"""Test c2d backbone."""
input_shape = (1, 3, 8, 64, 64)
imgs = generate_backbone_demo_inputs(input_shape)
# c2d inference test
c2d_r50 = C2D(depth=50)
c2d_r50.init_weights()
c2d_r50.train()
feat = c2d_r50(imgs)
assert feat.shape == torch.Size([1, 2048, 4, 2, 2])
c2d_r101 = C2D(depth=101)
c2d_r101.init_weights()
c2d_r101.train()
feat = c2d_r101(imgs)
assert feat.shape == torch.Size([1, 2048, 4, 2, 2])
|