vendor/ViTPose/tests/test_backbones/test_v2v_net.py

# Copyright (c) OpenMMLab. All rights reserved.
import torch

from mmpose.models import builder


def test_v2v_net():
    """Test V2VNet."""
    cfg = dict(type='V2VNet', input_channels=17, output_channels=15),
    model = builder.build_backbone(*cfg)
    input = torch.randn(2, 17, 32, 32, 32)
    output = model(input)
    assert output.shape == (2, 15, 32, 32, 32)