mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

update mmpose (#1213)

Browse Source
This commit is contained in:
RunningLeon 2022-10-18 10:24:19 +08:00 committed by GitHub
parent 86da0df105
commit 72923e7844
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements/codebases.txt
View File

@ -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

6
tests/test_codebase/test_mmpose/test_mmpose_models.py
View File

@ -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