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)
|