|
|
|
|
|
import pytest
|
|
|
import torch
|
|
|
|
|
|
from mmaction.models import RGBPoseConv3D
|
|
|
from mmaction.testing import generate_backbone_demo_inputs
|
|
|
|
|
|
|
|
|
def test_rgbposeconv3d():
|
|
|
"""Test RGBPoseConv3D backbone."""
|
|
|
|
|
|
with pytest.raises(AssertionError):
|
|
|
RGBPoseConv3D(pose_drop_path=1.1, rgb_drop_path=1.1)
|
|
|
|
|
|
rgbposec3d = RGBPoseConv3D()
|
|
|
rgbposec3d.init_weights()
|
|
|
rgbposec3d.train()
|
|
|
|
|
|
imgs_shape = (1, 3, 8, 224, 224)
|
|
|
heatmap_imgs_shape = (1, 17, 32, 56, 56)
|
|
|
imgs = generate_backbone_demo_inputs(imgs_shape)
|
|
|
heatmap_imgs = generate_backbone_demo_inputs(heatmap_imgs_shape)
|
|
|
|
|
|
(x_rgb, x_pose) = rgbposec3d(imgs, heatmap_imgs)
|
|
|
|
|
|
assert x_rgb.shape == torch.Size([1, 2048, 8, 7, 7])
|
|
|
assert x_pose.shape == torch.Size([1, 512, 32, 7, 7])
|
|
|
|
