From 72923e78448e65aa996e2fdcb1c0aa31b4553e2f Mon Sep 17 00:00:00 2001 From: RunningLeon Date: Tue, 18 Oct 2022 10:24:19 +0800 Subject: [PATCH] update mmpose (#1213) --- requirements/codebases.txt | 2 +- tests/test_codebase/test_mmpose/test_mmpose_models.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) 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