mirror of https://github.com/open-mmlab/mmocr.git
52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
import os.path as osp
|
|||
|
|
import re
|
|||
|
|
|
|||
|
|
# This script reads /projects/selected.txt and generate projectzoo.md
|
|||
|
|
|
|||
|
|
files = []
|
|||
|
|
|
|||
|
|
project_zoo = """
|
|||
|
|
# 前沿模型
|
|||
|
|
|
|||
|
|
这里是一些已经复现，但是尚未包含在 MMOCR 包中的前沿模型。
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
files = open('../../projects/selected.txt').readlines()
|
|||
|
|
|
|||
|
|
for file in files:
|
|||
|
|
file = file.strip()
|
|||
|
|
with open(osp.join('../../', file)) as f:
|
|||
|
|
content = f.read()
|
|||
|
|
|
|||
|
|
# Extract title
|
|||
|
|
expr = '# (.*?)\n'
|
|||
|
|
title = re.search(expr, content).group(1)
|
|||
|
|
project_zoo += f'## {title}\n\n'
|
|||
|
|
|
|||
|
|
# Locate the description
|
|||
|
|
expr = '## Description\n(.*?)##'
|
|||
|
|
description = re.search(expr, content, re.DOTALL).group(1)
|
|||
|
|
project_zoo += f'{description}\n'
|
|||
|
|
|
|||
|
|
# check milestone 1
|
|||
|
|
expr = r'- \[(.?)\] Milestone 1'
|
|||
|
|
state = re.search(expr, content, re.DOTALL).group(1)
|
|||
|
|
infer_state = '✔' if state == 'x' else '❌'
|
|||
|
|
|
|||
|
|
# check milestone 2
|
|||
|
|
expr = r'- \[(.?)\] Milestone 2'
|
|||
|
|
state = re.search(expr, content, re.DOTALL).group(1)
|
|||
|
|
training_state = '✔' if state == 'x' else '❌'
|
|||
|
|
|
|||
|
|
# add table
|
|||
|
|
readme_link = f'https://github.com/open-mmlab/mmocr/blob/dev-1.x/{file}'
|
|||
|
|
project_zoo += '### 模型状态 \n'
|
|||
|
|
project_zoo += '| 推理 | 训练 | README |\n'
|
|||
|
|
project_zoo += '| --------- | -------- | ------ |\n'
|
|||
|
|
project_zoo += f'|️{infer_state}|{training_state}|[link]({readme_link})|\n'
|
|||
|
|
|
|||
|
|
with open('projectzoo.md', 'w') as f:
|
|||
|
|
f.write(project_zoo)
|