model zoo

vernitgarg · Oct 8, 2019 · 551e3b4 · 551e3b4
1 parent d2604dd
commit 551e3b4
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/doc/MODEL_ZOO.md b/doc/MODEL_ZOO.md
@@ -1,2 +1,12 @@
 ## MODEL ZOO
-Coming Soon...
+<!-- Coming soon... -->
+
+<!-- We use AWS as the main site to host our model zoo, and maintain a mirror on aliyun. 
+You can replace https://s3.ap-northeast-2.amazonaws.com/open-mmlab with https://open-mmlab.oss-cn-beijing.aliyuncs.com in model urls. -->
+
+MMSkeleton usually automatically download necessary models from AWS in the runtime.
+
+We also maintain a mirror backup on [GoogleDrive](https://drive.google.com/open?id=1zC9ptIQTUoT7RvRM9Ec651cF5Xe7pty0)
+and [BaiduYun](https://pan.baidu.com/s/1iqOoQmIywuDQckgmehQ8HQ).
+As a plan B, you can manually download models and put them into checkpoints cache folder of pytorch.
+The folder defaults to ``~/.cache/torch/checkpoints`` in the Linux filesytem layout.
diff --git a/mmskeleton/utils/checkpoint.py b/mmskeleton/utils/checkpoint.py
@@ -2,12 +2,13 @@
 from mmcv.runner.checkpoint import load_url_dist
 import urllib
 
+
 mmskeleton_model_urls = {
     'st_gcn/kinetics-skeleton': "https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmskeleton/models/st-gcn/st_gcn.kinetics-6fa43f73.pth",
     'st_gcn/ntu-xsub': "https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmskeleton/models/st-gcn/st_gcn.ntu-xsub-300b57d4.pth",
     'st_gcn/ntu-xview': "https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmskeleton/models/st-gcn/st_gcn.ntu-xview-9ba67746.pth",
     'mmdet/htc_dconv_c3-c5_mstrain_400_1400_x101_64x4d_fpn_20e': 'https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmskeleton/models/mmdet/htc_dconv_c3-c5_mstrain_400_1400_x101_64x4d_fpn_20e_20190408-0e50669c.pth',
-    'pose_estimation/pose_hrnet_w32_256x192': 'https://1open-mmlab.s3.ap-northeast-2.amazonaws.com/mmskeleton/models/pose_estimation/pose_hrnet_w32_256x192-76ea353b.pth',
+    'pose_estimation/pose_hrnet_w32_256x192': 'https://open-mmlab.s3.ap-northeast-2.amazonaws.com/mmskeleton/models/pose_estimation/pose_hrnet_w32_256x192-76ea353b.pth',
     'mmdet/cascade_rcnn_r50_fpn_20e': 'https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/cascade_rcnn_r50_fpn_20e_20181123-db483a09.pth',
 }  # yapf: disable