mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update stat to classify papers (#139) 

* update stat to classify papers

* modify tag of papers
pull/140/head
congee 2021-01-14 16:48:05 +08:00 committed by GitHub
parent 63f38988eb
commit c8109d1eca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
10 changed files with 29 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/lenet/README.md
View File

@ -2,7 +2,7 @@
## Introduction ## Introduction
[BACKBONE] [ALGORITHM]
```latex ```latex
@ARTICLE{6795724, @ARTICLE{6795724,

2
configs/mobilenet_v2/README.md
View File

@ -2,7 +2,7 @@
## Introduction ## Introduction
[BACKBONE] [ALGORITHM]
```latex ```latex
@INPROCEEDINGS{8578572, @INPROCEEDINGS{8578572,

2
configs/resnet/README.md
View File

@ -2,7 +2,7 @@
## Introduction ## Introduction
[BACKBONE] [ALGORITHM]
```latex ```latex
@inproceedings{he2016deep, @inproceedings{he2016deep,

2
configs/resnext/README.md
View File

@ -2,7 +2,7 @@
## Introduction ## Introduction
[BACKBONE] [ALGORITHM]
```latex ```latex
@inproceedings{xie2017aggregated, @inproceedings{xie2017aggregated,

2
configs/seresnet/README.md
View File

@ -2,7 +2,7 @@
## Introduction ## Introduction
[BACKBONE] [ALGORITHM]
```latex ```latex
@inproceedings{hu2018squeeze, @inproceedings{hu2018squeeze,

2
configs/seresnext/README.md
View File

@ -2,7 +2,7 @@
## Introduction ## Introduction
[BACKBONE] [ALGORITHM]
```latex ```latex
@inproceedings{hu2018squeeze, @inproceedings{hu2018squeeze,

2
configs/shufflenet_v1/README.md
View File

@ -2,7 +2,7 @@
## Introduction ## Introduction
[BACKBONE] [ALGORITHM]
```latex ```latex
@inproceedings{zhang2018shufflenet, @inproceedings{zhang2018shufflenet,

2
configs/shufflenet_v2/README.md
View File

@ -2,7 +2,7 @@
## Introduction ## Introduction
[BACKBONE] [ALGORITHM]
```latex ```latex
@inproceedings{ma2018shufflenet, @inproceedings{ma2018shufflenet,

2
configs/vgg/README.md
View File

@ -2,7 +2,7 @@
## Introduction ## Introduction
[BACKBONE] [ALGORITHM]
```latex ```latex
@article{simonyan2014very, @article{simonyan2014very,

23
docs/stat.py
View File

@ -1,8 +1,11 @@
#!/usr/bin/env python #!/usr/bin/env python
import functools as func
import glob import glob
import os.path as osp import os.path as osp
import re import re
import numpy as np
url_prefix = 'https://github.com/open-mmlab/mmclassification/blob/master/' url_prefix = 'https://github.com/open-mmlab/mmclassification/blob/master/'
files = sorted(glob.glob('../configs/*/README.md')) files = sorted(glob.glob('../configs/*/README.md'))
@ -18,7 +21,7 @@ for f in files:
with open(f, 'r') as content_file: with open(f, 'r') as content_file:
content = content_file.read() content = content_file.read()
title = content.split('\n')[0].replace('# ', '') title = content.split('\n')[0].replace('# ', '').strip()
ckpts = set(x.lower().strip() ckpts = set(x.lower().strip()
for x in re.findall(r'\[model\]\((https?.*)\)', content)) for x in re.findall(r'\[model\]\((https?.*)\)', content))
@ -26,20 +29,34 @@ for f in files:
if len(ckpts) == 0: if len(ckpts) == 0:
continue continue
_papertype = [x for x in re.findall(r'\[([A-Z]+)\]', content)]
assert len(_papertype) > 0
papertype = _papertype[0]
paper = set([(papertype, title)])
num_ckpts += len(ckpts) num_ckpts += len(ckpts)
titles.append(title) titles.append(title)
statsmsg = f""" statsmsg = f"""
\t* [{title}]({url}) ({len(ckpts)} ckpts) \t* [{papertype}] [{title}]({url}) ({len(ckpts)} ckpts)
""" """
stats.append((title, ckpts, statsmsg)) stats.append((paper, ckpts, statsmsg))
allpapers = func.reduce(lambda a, b: a.union(b), [p for p, _, _ in stats])
msglist = '\n'.join(x for _, _, x in stats) msglist = '\n'.join(x for _, _, x in stats)
papertypes, papercounts = np.unique([t for t, _ in allpapers],
return_counts=True)
countstr = '\n'.join(
[f' - {t}: {c}' for t, c in zip(papertypes, papercounts)])
modelzoo = f""" modelzoo = f"""
# Model Zoo Statistics # Model Zoo Statistics
* Number of papers: {len(set(titles))} * Number of papers: {len(set(titles))}
{countstr}
* Number of checkpoints: {num_ckpts} * Number of checkpoints: {num_ckpts}
{msglist} {msglist}
""" """