[Fix] fix model zoo collection (open-mmlab#1655)

wjkim81 · Sep 14, 2022 · 19d20a7 · 19d20a7
1 parent 9e71b78
commit 19d20a7
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 8 deletions.
diff --git a/docs/en/collect_modelzoo.py b/docs/en/collect_modelzoo.py
@@ -58,7 +58,7 @@ def _titlecase_callback(word, **kwargs):
     dataset = _rel_path[3]
 
     # get keywords
-    keywords_algo = (_rel_path[1], )
+    keywords_algo = (_rel_path[2], )
     keywords_setting = tuple(_rel_path[4][:-3].split('_'))
     keywords = keywords_algo + keywords_setting
 
@@ -137,10 +137,14 @@ def main():
                 keyword_strs = [
                     titlecase(x.replace('_', ' ')) for x in keywords
                 ]
+                dataset_str = titlecase(dataset)
+                if dataset_str in keyword_strs:
+                    keyword_strs.remove(dataset_str)
+
                 lines += [
                     '<br/>', '',
                     (f'### {" + ".join(keyword_strs)}'
-                     f' on {titlecase(dataset)}'), '', doc['content'], ''
+                     f' on {dataset_str}'), '', doc['content'], ''
                 ]
 
         fn = osp.join('model_zoo', f'{task.replace(" ", "_").lower()}.md')
@@ -165,11 +169,14 @@ def main():
                         keyword_strs = [
                             titlecase(x.replace('_', ' ')) for x in keywords
                         ]
+
+                        dataset_str = titlecase(dataset)
+                        if dataset_str in keyword_strs:
+                            keyword_strs.remove(dataset_str)
                         paperlines += [
                             '<br/>', '',
                             (f'### {" + ".join(keyword_strs)}'
-                             f' on {titlecase(dataset)}'), '',
-                            doc_info['content'], ''
+                             f' on {dataset_str}'), '', doc_info['content'], ''
                         ]
             if paperlines:
                 lines += ['<hr/>', '<br/><br/>', '', f'## {paper_name}', '']

diff --git a/docs/src/papers/algorithms/simcc.md b/docs/src/papers/algorithms/simcc.md
diff --git a/docs/zh_cn/collect_modelzoo.py b/docs/zh_cn/collect_modelzoo.py
@@ -58,7 +58,7 @@ def _titlecase_callback(word, **kwargs):
     dataset = _rel_path[3]
 
     # get keywords
-    keywords_algo = (_rel_path[1], )
+    keywords_algo = (_rel_path[2], )
     keywords_setting = tuple(_rel_path[4][:-3].split('_'))
     keywords = keywords_algo + keywords_setting
 
@@ -137,10 +137,14 @@ def main():
                 keyword_strs = [
                     titlecase(x.replace('_', ' ')) for x in keywords
                 ]
+                dataset_str = titlecase(dataset)
+                if dataset_str in keyword_strs:
+                    keyword_strs.remove(dataset_str)
+
                 lines += [
                     '<br/>', '',
                     (f'### {" + ".join(keyword_strs)}'
-                     f' on {titlecase(dataset)}'), '', doc['content'], ''
+                     f' on {dataset_str}'), '', doc['content'], ''
                 ]
 
         fn = osp.join('model_zoo', f'{task.replace(" ", "_").lower()}.md')
@@ -165,11 +169,14 @@ def main():
                         keyword_strs = [
                             titlecase(x.replace('_', ' ')) for x in keywords
                         ]
+
+                        dataset_str = titlecase(dataset)
+                        if dataset_str in keyword_strs:
+                            keyword_strs.remove(dataset_str)
                         paperlines += [
                             '<br/>', '',
                             (f'### {" + ".join(keyword_strs)}'
-                             f' on {titlecase(dataset)}'), '',
-                            doc_info['content'], ''
+                             f' on {dataset_str}'), '', doc_info['content'], ''
                         ]
             if paperlines:
                 lines += ['<hr/>', '<br/><br/>', '', f'## {paper_name}', '']