26 lines
627 B
Python
26 lines
627 B
Python
import json
|
|
from pathlib import Path
|
|
|
|
from tqdm import tqdm
|
|
|
|
from manga_ocr import MangaOcr
|
|
|
|
TEST_DATA_ROOT = Path(__file__).parent / 'data'
|
|
|
|
|
|
def generate_expected_results():
|
|
mocr = MangaOcr()
|
|
|
|
results = []
|
|
|
|
for path in tqdm(sorted((TEST_DATA_ROOT / 'images').iterdir())):
|
|
result = mocr(path)
|
|
results.append({'filename': path.name, 'result': result})
|
|
|
|
(TEST_DATA_ROOT / 'expected_results.json').write_text(json.dumps(results, ensure_ascii=False, indent=2),
|
|
encoding='utf-8')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
generate_expected_results()
|