From 19d20a7a0c3f742bdb89a9aef3115be17adf6af1 Mon Sep 17 00:00:00 2001
From: Yining Li <liyining0712@gmail.com>
Date: Wed, 14 Sep 2022 10:57:47 +0800
Subject: [PATCH] [Fix] fix model zoo collection (#1655)

---
 docs/en/collect_modelzoo.py         | 15 +++++++++++----
 docs/src/papers/algorithms/simcc.md |  0
 docs/zh_cn/collect_modelzoo.py      | 15 +++++++++++----
 3 files changed, 22 insertions(+), 8 deletions(-)
 delete mode 100644 docs/src/papers/algorithms/simcc.md

diff --git a/docs/en/collect_modelzoo.py b/docs/en/collect_modelzoo.py
index 264b18d849e..0c87d3c6ef6 100644
--- 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
deleted file mode 100644
index e69de29bb2d..00000000000
diff --git a/docs/zh_cn/collect_modelzoo.py b/docs/zh_cn/collect_modelzoo.py
index 264b18d849e..0c87d3c6ef6 100644
--- 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}', '']