open-mmlab · ly015 · Nov 25, 2021 · Nov 2, 2021
diff --git a/.dev_scripts/github/update_model_index.py b/.dev_scripts/github/update_model_index.py
@@ -149,16 +149,15 @@ def parse_md(md_file):
     """
     readme_path = osp.relpath(md_file, MMPOSE_ROOT)
 
-    collection = dict(
-        Name=None,  # fill in later
-        Metadata={'Architecture': []},
-        README=readme_path,
-        Paper=None)
+    collection = {'Name': None, 'Paper': None}
     models = []
 
     # record the publish year of the latest paper
     paper_year = -1
-    dataset = 'Unknown'
+    dataset = None
+
+    # architectures for collection and model
+    architecture = []
 
     with open(md_file, 'r') as md:
         lines = md.readlines()
@@ -180,17 +179,19 @@ def parse_md(md_file):
 
                 paper_type = re.findall(r'\[(.*)\]', lines[i])[0]
 
+                # lower priority for dataset paper
                 if paper_type == 'DATASET':
-                    # DATASET paper has lower priority
                     year = 0
 
                 if year > paper_year:
                     collection['Paper'] = dict(Title=title, URL=url)
+                    collection['Name'] = name
                     paper_year = year
 
                 # get architecture
                 if paper_type in {'ALGORITHM', 'BACKBONE'}:
-                    collection['Metadata']['Architecture'].append(name)
+                    architecture.append(name)
+
                 # get dataset
                 elif paper_type == 'DATASET':
                     dataset = name
@@ -234,7 +235,10 @@ def parse_md(md_file):
                     task_name = ' '.join(
                         [model_info['target'], model_info['task']])
 
-                    metadata = {'Training Data': dataset}
+                    metadata = {
+                        'Training Data': dataset,
+                        'Architecture': architecture
+                    }
                     if flops_idx != -1:
                         metadata['FLOPs'] = float(line[flops_idx])
                     if params_idx != -1:
@@ -248,8 +252,10 @@ def parse_md(md_file):
                     model = {
                         'Name':
                         model_name,
+                        'README':
+                        readme_path,
                         'In Collection':
-                        None,  # fill in later
+                        collection['Name'],
                         'Config':
                         config,
                         'Metadata':
@@ -269,19 +275,8 @@ def parse_md(md_file):
             else:
                 i += 1
 
-    # fill in collection name
-    readme_name = '--'.join(
-        osp.splitext(osp.relpath(readme_path, 'configs'))[0].split(osp.sep))
-    collection_alias = '+'.join(
-        collection['Metadata']['Architecture']) + f'@{dataset}'
-    collection_name = f'{readme_name} [{collection_alias}]'
-    collection['Name'] = collection_name
-    for model in models:
-        model['In Collection'] = collection_name
-
     result = {'Collections': [collection], 'Models': models}
-    yml_file = md_file[:-2] + 'yml'
-
+    yml_file = osp.splitext(md_file)[0] + '.yml'
     is_different = dump_yaml_and_check_difference(result, yml_file)
     return is_different
 

diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_animalpose.yml b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_animalpose.yml
@@ -1,20 +1,17 @@
 Collections:
-- Metadata:
-    Architecture:
-    - HRNet
-  Name: animal--2d_kpt_sview_rgb_img--topdown_heatmap--animalpose--hrnet_animalpose
-    [HRNet@Animal-Pose]
+- Name: HRNet
   Paper:
     Title: Deep high-resolution representation learning for human pose estimation
     URL: http://openaccess.thecvf.com/content_CVPR_2019/html/Sun_Deep_High-Resolution_Representation_Learning_for_Human_Pose_Estimation_CVPR_2019_paper.html
-  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_animalpose.md
 Models:
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_w32_animalpose_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--animalpose--hrnet_animalpose
-    [HRNet@Animal-Pose]
+  In Collection: HRNet
   Metadata:
+    Architecture: &id001
+    - HRNet
     Training Data: Animal-Pose
   Name: topdown_heatmap_hrnet_w32_animalpose_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_animalpose.md
   Results:
   - Dataset: Animal-Pose
     Metrics:
@@ -26,11 +23,12 @@ Models:
     Task: Animal 2D Keypoint
   Weights: https://download.openmmlab.com/mmpose/animal/hrnet/hrnet_w32_animalpose_256x256-1aa7f075_20210426.pth
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_w48_animalpose_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--animalpose--hrnet_animalpose
-    [HRNet@Animal-Pose]
+  In Collection: HRNet
   Metadata:
+    Architecture: *id001
     Training Data: Animal-Pose
   Name: topdown_heatmap_hrnet_w48_animalpose_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_animalpose.md
   Results:
   - Dataset: Animal-Pose
     Metrics:

diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/resnet_animalpose.yml b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/resnet_animalpose.yml
@@ -1,20 +1,17 @@
 Collections:
-- Metadata:
-    Architecture:
-    - SimpleBaseline2D
-  Name: animal--2d_kpt_sview_rgb_img--topdown_heatmap--animalpose--resnet_animalpose
-    [SimpleBaseline2D@Animal-Pose]
+- Name: SimpleBaseline2D
   Paper:
     Title: Simple baselines for human pose estimation and tracking
     URL: http://openaccess.thecvf.com/content_ECCV_2018/html/Bin_Xiao_Simple_Baselines_for_ECCV_2018_paper.html
-  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/resnet_animalpose.md
 Models:
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res50_animalpose_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--animalpose--resnet_animalpose
-    [SimpleBaseline2D@Animal-Pose]
+  In Collection: SimpleBaseline2D
   Metadata:
+    Architecture: &id001
+    - SimpleBaseline2D
     Training Data: Animal-Pose
   Name: topdown_heatmap_res50_animalpose_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/resnet_animalpose.md
   Results:
   - Dataset: Animal-Pose
     Metrics:
@@ -26,11 +23,12 @@ Models:
     Task: Animal 2D Keypoint
   Weights: https://download.openmmlab.com/mmpose/animal/resnet/res50_animalpose_256x256-e1f30bff_20210426.pth
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res101_animalpose_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--animalpose--resnet_animalpose
-    [SimpleBaseline2D@Animal-Pose]
+  In Collection: SimpleBaseline2D
   Metadata:
+    Architecture: *id001
     Training Data: Animal-Pose
   Name: topdown_heatmap_res101_animalpose_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/resnet_animalpose.md
   Results:
   - Dataset: Animal-Pose
     Metrics:
@@ -42,11 +40,12 @@ Models:
     Task: Animal 2D Keypoint
   Weights: https://download.openmmlab.com/mmpose/animal/resnet/res101_animalpose_256x256-85563f4a_20210426.pth
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res152_animalpose_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--animalpose--resnet_animalpose
-    [SimpleBaseline2D@Animal-Pose]
+  In Collection: SimpleBaseline2D
   Metadata:
+    Architecture: *id001
     Training Data: Animal-Pose
   Name: topdown_heatmap_res152_animalpose_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/resnet_animalpose.md
   Results:
   - Dataset: Animal-Pose
     Metrics:

diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/hrnet_ap10k.yml b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/hrnet_ap10k.yml
@@ -1,19 +1,17 @@
 Collections:
-- Metadata:
-    Architecture:
-    - HRNet
-  Name: animal--2d_kpt_sview_rgb_img--topdown_heatmap--ap10k--hrnet_ap10k [HRNet@AP-10K]
+- Name: HRNet
   Paper:
     Title: Deep high-resolution representation learning for human pose estimation
     URL: http://openaccess.thecvf.com/content_CVPR_2019/html/Sun_Deep_High-Resolution_Representation_Learning_for_Human_Pose_Estimation_CVPR_2019_paper.html
-  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/hrnet_ap10k.md
 Models:
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/hrnet_w32_ap10k_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--ap10k--hrnet_ap10k
-    [HRNet@AP-10K]
+  In Collection: HRNet
   Metadata:
+    Architecture: &id001
+    - HRNet
     Training Data: AP-10K
   Name: topdown_heatmap_hrnet_w32_ap10k_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/hrnet_ap10k.md
   Results:
   - Dataset: AP-10K
     Metrics:
@@ -25,11 +23,12 @@ Models:
     Task: Animal 2D Keypoint
   Weights: https://download.openmmlab.com/mmpose/animal/hrnet/hrnet_w32_ap10k_256x256-18aac840_20211029.pth
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/hrnet_w48_ap10k_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--ap10k--hrnet_ap10k
-    [HRNet@AP-10K]
+  In Collection: HRNet
   Metadata:
+    Architecture: *id001
     Training Data: AP-10K
   Name: topdown_heatmap_hrnet_w48_ap10k_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/hrnet_ap10k.md
   Results:
   - Dataset: AP-10K
     Metrics:

diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/resnet_ap10k.yml b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/resnet_ap10k.yml
@@ -1,19 +1,17 @@
 Collections:
-- Metadata:
-    Architecture:
-    - SimpleBaseline2D
-  Name: animal--2d_kpt_sview_rgb_img--topdown_heatmap--ap10k--resnet_ap10k [SimpleBaseline2D@AP-10K]
+- Name: SimpleBaseline2D
   Paper:
     Title: Simple baselines for human pose estimation and tracking
     URL: http://openaccess.thecvf.com/content_ECCV_2018/html/Bin_Xiao_Simple_Baselines_for_ECCV_2018_paper.html
-  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/resnet_ap10k.md
 Models:
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/res50_ap10k_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--ap10k--resnet_ap10k
-    [SimpleBaseline2D@AP-10K]
+  In Collection: SimpleBaseline2D
   Metadata:
+    Architecture: &id001
+    - SimpleBaseline2D
     Training Data: AP-10K
   Name: topdown_heatmap_res50_ap10k_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/resnet_ap10k.md
   Results:
   - Dataset: AP-10K
     Metrics:
@@ -25,11 +23,12 @@ Models:
     Task: Animal 2D Keypoint
   Weights: https://download.openmmlab.com/mmpose/animal/resnet/res50_ap10k_256x256-35760eb8_20211029.pth
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/res101_ap10k_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--ap10k--resnet_ap10k
-    [SimpleBaseline2D@AP-10K]
+  In Collection: SimpleBaseline2D
   Metadata:
+    Architecture: *id001
     Training Data: AP-10K
   Name: topdown_heatmap_res101_ap10k_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/ap10k/resnet_ap10k.md
   Results:
   - Dataset: AP-10K
     Metrics:

diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_atrw.yml b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_atrw.yml
@@ -1,18 +1,17 @@
 Collections:
-- Metadata:
-    Architecture:
-    - HRNet
-  Name: animal--2d_kpt_sview_rgb_img--topdown_heatmap--atrw--hrnet_atrw [HRNet@ATRW]
+- Name: HRNet
   Paper:
     Title: Deep high-resolution representation learning for human pose estimation
     URL: http://openaccess.thecvf.com/content_CVPR_2019/html/Sun_Deep_High-Resolution_Representation_Learning_for_Human_Pose_Estimation_CVPR_2019_paper.html
-  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_atrw.md
 Models:
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_w32_atrw_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--atrw--hrnet_atrw [HRNet@ATRW]
+  In Collection: HRNet
   Metadata:
+    Architecture: &id001
+    - HRNet
     Training Data: ATRW
   Name: topdown_heatmap_hrnet_w32_atrw_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_atrw.md
   Results:
   - Dataset: ATRW
     Metrics:
@@ -24,10 +23,12 @@ Models:
     Task: Animal 2D Keypoint
   Weights: https://download.openmmlab.com/mmpose/animal/hrnet/hrnet_w32_atrw_256x256-f027f09a_20210414.pth
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_w48_atrw_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--atrw--hrnet_atrw [HRNet@ATRW]
+  In Collection: HRNet
   Metadata:
+    Architecture: *id001
     Training Data: ATRW
   Name: topdown_heatmap_hrnet_w48_atrw_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_atrw.md
   Results:
   - Dataset: ATRW
     Metrics:

diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/resnet_atrw.yml b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/resnet_atrw.yml
@@ -1,19 +1,17 @@
 Collections:
-- Metadata:
-    Architecture:
-    - SimpleBaseline2D
-  Name: animal--2d_kpt_sview_rgb_img--topdown_heatmap--atrw--resnet_atrw [SimpleBaseline2D@ATRW]
+- Name: SimpleBaseline2D
   Paper:
     Title: Simple baselines for human pose estimation and tracking
     URL: http://openaccess.thecvf.com/content_ECCV_2018/html/Bin_Xiao_Simple_Baselines_for_ECCV_2018_paper.html
-  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/resnet_atrw.md
 Models:
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res50_atrw_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--atrw--resnet_atrw
-    [SimpleBaseline2D@ATRW]
+  In Collection: SimpleBaseline2D
   Metadata:
+    Architecture: &id001
+    - SimpleBaseline2D
     Training Data: ATRW
   Name: topdown_heatmap_res50_atrw_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/resnet_atrw.md
   Results:
   - Dataset: ATRW
     Metrics:
@@ -25,11 +23,12 @@ Models:
     Task: Animal 2D Keypoint
   Weights: https://download.openmmlab.com/mmpose/animal/resnet/res50_atrw_256x256-546c4594_20210414.pth
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res101_atrw_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--atrw--resnet_atrw
-    [SimpleBaseline2D@ATRW]
+  In Collection: SimpleBaseline2D
   Metadata:
+    Architecture: *id001
     Training Data: ATRW
   Name: topdown_heatmap_res101_atrw_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/resnet_atrw.md
   Results:
   - Dataset: ATRW
     Metrics:
@@ -41,11 +40,12 @@ Models:
     Task: Animal 2D Keypoint
   Weights: https://download.openmmlab.com/mmpose/animal/resnet/res101_atrw_256x256-da93f371_20210414.pth
 - Config: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res152_atrw_256x256.py
-  In Collection: animal--2d_kpt_sview_rgb_img--topdown_heatmap--atrw--resnet_atrw
-    [SimpleBaseline2D@ATRW]
+  In Collection: SimpleBaseline2D
   Metadata:
+    Architecture: *id001
     Training Data: ATRW
   Name: topdown_heatmap_res152_atrw_256x256
+  README: configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/resnet_atrw.md
   Results:
   - Dataset: ATRW
     Metrics: