You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
13 lines
368 B
13 lines
368 B
# 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)
|
|
|