skip voxelpose unittest for Windows

open-mmlab · Mar 4, 2022 · 5c1ddab · 5c1ddab
1 parent 07f2600
commit 5c1ddab
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/configs/_base_/filters/gausian_filter.py b/configs/_base_/filters/gausian_filter.py
diff --git a/tests/test_models/test_voxelpose.py b/tests/test_models/test_voxelpose.py
@@ -1,6 +1,8 @@
 # Copyright (c) OpenMMLab. All rights reserved.
+import platform
 import tempfile
 
+import pytest
 from mmcv import Config
 
 from mmpose.datasets import DATASETS, build_dataloader
@@ -9,6 +11,10 @@
                                                PoseRegressionNet, ProjectLayer)
 
 
+@pytest.mark.skipif(
+    platform.system == 'Windows',
+    reason='Windows container on Github Action does not have enough RAM '
+    'to run the test.')
 def test_voxelpose_forward():
     dataset = 'Body3DMviewDirectPanopticDataset'
     dataset_class = DATASETS.get(dataset)
@@ -18,8 +24,8 @@ def test_voxelpose_forward():
     space_center = [0, -500, 800]
     cube_size = [20, 20, 8]
     train_data_cfg = dict(
-        image_size=[240, 128],
-        heatmap_size=[[60, 32]],
+        image_size=[960, 512],
+        heatmap_size=[[240, 128]],
         space_size=space_size,
         space_center=space_center,
         cube_size=cube_size,
@@ -58,7 +64,7 @@ def test_voxelpose_forward():
     ]
 
     project_layer = ProjectLayer(
-        dict(image_size=[240, 128], heatmap_size=[60, 32]))
+        dict(image_size=[960, 512], heatmap_size=[240, 128]))
     root_net = CuboidProposalNet(
         dict(type='V2VNet', input_channels=15, output_channels=1))
     center_head = builder.build_head(