diff --git a/requirements/codebases.txt b/requirements/codebases.txt index 035febdb1..d08a2d4c5 100644 --- a/requirements/codebases.txt +++ b/requirements/codebases.txt @@ -2,5 +2,5 @@ mmcls>=1.0.0rc2 mmdet @ git+https://github.com/open-mmlab/mmdetection.git@dev-3.x mmedit @ git+https://github.com/open-mmlab/mmediting.git@1.x mmocr @ git+https://github.com/open-mmlab/mmocr.git@dev-1.x -mmpose>=1.0.0b0 +mmpose>=1.0.0rc0 mmsegmentation>=1.0.0rc0 diff --git a/tests/test_codebase/test_mmpose/test_mmpose_models.py b/tests/test_codebase/test_mmpose/test_mmpose_models.py index 2f544a5dd..eecf9226c 100644 --- a/tests/test_codebase/test_mmpose/test_mmpose_models.py +++ b/tests/test_codebase/test_mmpose/test_mmpose_models.py @@ -49,7 +49,11 @@ def test_heatmaphead_predict(backend_type: Backend): def get_msmu_head(): from mmpose.models.heads import MSPNHead model = MSPNHead( - num_stages=1, num_units=1, out_shape=(32, 48), unit_channels=16) + num_stages=1, + num_units=1, + out_shape=(32, 48), + unit_channels=16, + level_indices=[1]) model.requires_grad_(False) return model