Skip to content

Commit c53336e

Browse files
authored
[Fix] Fix the error when downloading config and checkpoint using mim (#2921)
1 parent 509441e commit c53336e

File tree

4 files changed

+42
-2
lines changed

4 files changed

+42
-2
lines changed

configs/face_2d_keypoint/topdown_regression/wflw/resnet_softwingloss_wflw.yml

+7-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,12 @@
1+
Collections:
2+
- Name: SoftWingloss
3+
Paper:
4+
Title: Structure-Coherent Deep Feature Learning for Robust Face Alignment
5+
URL: https://ieeexplore.ieee.org/document/9442331/
6+
README: https://github.com/open-mmlab/mmpose/blob/main/docs/src/papers/techniques/softwingloss.md
17
Models:
28
- Config: configs/face_2d_keypoint/topdown_regression/wflw/td-reg_res50_softwingloss_8xb64-210e_wflw-256x256.py
3-
In Collection: ResNet
9+
In Collection: SoftWingloss
410
Metadata:
511
Architecture:
612
- DeepPose

configs/face_2d_keypoint/topdown_regression/wflw/resnet_wflw.yml

+6
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,9 @@
1+
Collections:
2+
- Name: ResNet
3+
Paper:
4+
Title: Deep residual learning for image recognition
5+
URL: http://openaccess.thecvf.com/content_cvpr_2016/html/He_Deep_Residual_Learning_CVPR_2016_paper.html
6+
README: https://github.com/open-mmlab/mmpose/blob/main/docs/src/papers/backbones/resnet.md
17
Models:
28
- Config: configs/face_2d_keypoint/topdown_regression/wflw/td-reg_res50_8xb64-210e_wflw-256x256.py
39
In Collection: ResNet

configs/face_2d_keypoint/topdown_regression/wflw/resnet_wingloss_wflw.yml

+8-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,13 @@
1+
Collections:
2+
- Name: Wingloss
3+
Paper:
4+
Title: Wing Loss for Robust Facial Landmark Localisation with Convolutional Neural
5+
Networks
6+
URL: http://openaccess.thecvf.com/content_cvpr_2018/html/Feng_Wing_Loss_for_CVPR_2018_paper.html
7+
README: https://github.com/open-mmlab/mmpose/blob/main/docs/src/papers/techniques/wingloss.md
18
Models:
29
- Config: configs/face_2d_keypoint/topdown_regression/wflw/td-reg_res50_wingloss_8xb64-210e_wflw-256x256.py
3-
In Collection: ResNet
10+
In Collection: Wingloss
411
Metadata:
512
Architecture:
613
- DeepPose

tests/test_external/test_mim.py

+21
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
# Copyright (c) OpenMMLab. All rights reserved.
2+
import os
3+
from tempfile import TemporaryDirectory
4+
from unittest import TestCase
5+
6+
from mim.commands import download
7+
8+
9+
class TestMIM(TestCase):
10+
11+
def test_download(self):
12+
with TemporaryDirectory() as tmp_dir:
13+
ckpts = download(
14+
'mmpose',
15+
configs=['td-hm_hrnet-w48_8xb32-210e_coco-256x192'],
16+
dest_root=tmp_dir)
17+
18+
self.assertEqual(len(ckpts), 1)
19+
self.assertIn('td-hm_hrnet-w48_8xb32-210e_coco-256x192.py',
20+
os.listdir(tmp_dir))
21+
self.assertIn(ckpts[0], os.listdir(tmp_dir))

0 commit comments

Comments
 (0)