s444415/donut
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

config changes

Browse Source
This commit is contained in:
s444415 2022-12-16 14:16:46 +00:00
parent 8ccd1aabb6
commit b7296bb2a9
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
config.yaml
View File

@ -1,7 +1,8 @@
pretrained_processor_path: "Zombely/plwiki-proto-fine-tuned-v2"
pretrained_model_path: "Zombely/plwiki-proto-fine-tuned-v2"
pretrained_processor_path: "naver-clova-ix/donut-proto"
pretrained_model_path: "naver-clova-ix/donut-proto"
validation_dataset_path: "Zombely/diachronia-ocr"
validation_dataset_split: "train"
has_metadata: False
print_output: True
output_file_dir: "../../gonito-outs"
test_name: "proto-test"

2
donut-eval.py
View File

@ -63,7 +63,7 @@ def main(config):
output_list.append(seq)
if config.output_file_dir:
df = pd.DataFrame(map(lambda x: x.get('text_sequence', ''), output_list))
df.to_csv(f'{config.output_file_dir}/{config.pretrained_processor_path}-out.tsv', sep='\t', header=False, index=False)
df.to_csv(f'{config.output_file_dir}/{config.test_name}-out.tsv', sep='\t', header=False, index=False)
if config.has_metadata:
scores = {"accuracies": accs, "mean_accuracy": np.mean(accs)}