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: ...@@ -18,10 +18,13 @@ for f in files:
content = content_file.read() content = content_file.read()
title = content.split('\n')[0].replace('#', '') title = content.split('\n')[0].replace('#', '')
titles.append(title)
ckpts = set(x.lower().strip() ckpts = set(x.lower().strip()
for x in re.findall(r'https?://download.*\.pth', content) for x in re.findall(r'https?://download.*\.pth', content)
if 'mmdetection3d' in x) if 'mmdetection3d' in x)
if len(ckpts) == 0:
continue
titles.append(title)
num_ckpts += len(ckpts) num_ckpts += len(ckpts)
statsmsg = f""" statsmsg = f"""
\t* [{title}]({url}) ({len(ckpts)} ckpts) \t* [{title}]({url}) ({len(ckpts)} ckpts)
...@@ -33,7 +36,7 @@ msglist = '\n'.join(x for _, _, x in stats) ...@@ -33,7 +36,7 @@ msglist = '\n'.join(x for _, _, x in stats)
modelzoo = f""" modelzoo = f"""
\n## Model Zoo Statistics \n## Model Zoo Statistics
* Number of papers: {len(titles)} * Number of papers: {len(set(titles))}
* Number of checkpoints: {num_ckpts} * Number of checkpoints: {num_ckpts}
{msglist} {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