File tree Expand file tree Collapse file tree 1 file changed +9
-2
lines changed Expand file tree Collapse file tree 1 file changed +9
-2
lines changed Original file line number Diff line number Diff line change 44import re
55
66url_prefix = 'https://github.com/open-mmlab/mmsegmentation/blob/master/'
7+ titles_to_be_excluded = ['Mixed Precision Training' ]
78
89files = sorted (glob .glob ('../configs/*/README.md' ))
910
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)
3340modelzoo = 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"""
You can’t perform that action at this time.
0 commit comments