Unverified Commit a79f464e authored by congee's avatar congee Committed by GitHub
Browse files

remove the models with 0 ckpt (#259)

parent d0068487
......@@ -18,10 +18,13 @@ for f in files:
content = content_file.read()
title = content.split('\n')[0].replace('#', '')
titles.append(title)
ckpts = set(x.lower().strip()
for x in re.findall(r'https?://download.*\.pth', content)
if 'mmdetection3d' in x)
if len(ckpts) == 0:
continue
titles.append(title)
num_ckpts += len(ckpts)
statsmsg = f"""
\t* [{title}]({url}) ({len(ckpts)} ckpts)
......@@ -33,7 +36,7 @@ msglist = '\n'.join(x for _, _, x in stats)
modelzoo = f"""
\n## Model Zoo Statistics
* Number of papers: {len(titles)}
* Number of papers: {len(set(titles))}
* Number of checkpoints: {num_ckpts}
{msglist}
"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment