Skip to content

Commit 8ed47ab

Browse files
authored
remove the model with 0 ckpts and exclude the papers without proposing any model (open-mmlab#336)
1 parent 77e8ce3 commit 8ed47ab

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

docs/stat.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import re
55

66
url_prefix = 'https://github.com/open-mmlab/mmsegmentation/blob/master/'
7+
titles_to_be_excluded = ['Mixed Precision Training']
78

89
files = sorted(glob.glob('../configs/*/README.md'))
910

@@ -18,10 +19,16 @@
1819
content = content_file.read()
1920

2021
title = content.split('\n')[0].replace('#', '')
21-
titles.append(title)
22+
if title.strip() in titles_to_be_excluded:
23+
continue
24+
2225
ckpts = set(x.lower().strip()
2326
for x in re.findall(r'https?://download.*\.pth', content)
2427
if 'mmsegmentation' in x)
28+
if len(ckpts) == 0:
29+
continue
30+
31+
titles.append(title)
2532
num_ckpts += len(ckpts)
2633
statsmsg = f"""
2734
\t* [{title}]({url}) ({len(ckpts)} ckpts)
@@ -33,7 +40,7 @@
3340
modelzoo = f"""
3441
# Model Zoo Statistics
3542
36-
* Number of papers: {len(titles)}
43+
* Number of papers: {len(set(titles))}
3744
* Number of checkpoints: {num_ckpts}
3845
{msglist}
3946
"""

0 commit comments

Comments
 (0)