Skip to content

Commit

Permalink
[Fix] Fix the error when downloading config and checkpoint using mim (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
Ben-Louis authored Jan 11, 2024
1 parent 509441e commit c53336e
Show file tree
Hide file tree
Showing 4 changed files with 42 additions and 2 deletions.
Original file line number Diff line number Diff line change
@@ -1,6 +1,12 @@
Collections:
- Name: SoftWingloss
Paper:
Title: Structure-Coherent Deep Feature Learning for Robust Face Alignment
URL: https://ieeexplore.ieee.org/document/9442331/
README: https://github.com/open-mmlab/mmpose/blob/main/docs/src/papers/techniques/softwingloss.md
Models:
- Config: configs/face_2d_keypoint/topdown_regression/wflw/td-reg_res50_softwingloss_8xb64-210e_wflw-256x256.py
In Collection: ResNet
In Collection: SoftWingloss
Metadata:
Architecture:
- DeepPose
Expand Down
Original file line number Diff line number Diff line change
@@ -1,3 +1,9 @@
Collections:
- Name: ResNet
Paper:
Title: Deep residual learning for image recognition
URL: http://openaccess.thecvf.com/content_cvpr_2016/html/He_Deep_Residual_Learning_CVPR_2016_paper.html
README: https://github.com/open-mmlab/mmpose/blob/main/docs/src/papers/backbones/resnet.md
Models:
- Config: configs/face_2d_keypoint/topdown_regression/wflw/td-reg_res50_8xb64-210e_wflw-256x256.py
In Collection: ResNet
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,13 @@
Collections:
- Name: Wingloss
Paper:
Title: Wing Loss for Robust Facial Landmark Localisation with Convolutional Neural
Networks
URL: http://openaccess.thecvf.com/content_cvpr_2018/html/Feng_Wing_Loss_for_CVPR_2018_paper.html
README: https://github.com/open-mmlab/mmpose/blob/main/docs/src/papers/techniques/wingloss.md
Models:
- Config: configs/face_2d_keypoint/topdown_regression/wflw/td-reg_res50_wingloss_8xb64-210e_wflw-256x256.py
In Collection: ResNet
In Collection: Wingloss
Metadata:
Architecture:
- DeepPose
Expand Down
21 changes: 21 additions & 0 deletions tests/test_external/test_mim.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
# Copyright (c) OpenMMLab. All rights reserved.
import os
from tempfile import TemporaryDirectory
from unittest import TestCase

from mim.commands import download


class TestMIM(TestCase):

def test_download(self):
with TemporaryDirectory() as tmp_dir:
ckpts = download(
'mmpose',
configs=['td-hm_hrnet-w48_8xb32-210e_coco-256x192'],
dest_root=tmp_dir)

self.assertEqual(len(ckpts), 1)
self.assertIn('td-hm_hrnet-w48_8xb32-210e_coco-256x192.py',
os.listdir(tmp_dir))
self.assertIn(ckpts[0], os.listdir(tmp_dir))

0 comments on commit c53336e

Please sign in to comment.