| """cutpreds.py <in.jsonl> <out.jsonl>: apply loopcut.cut_loops to recognizer outputs, keeping the raw output | |
| in pred_raw (the same transformation that produced preds/ocr_*_ocr4b_mixcut.jsonl).""" | |
| import json | |
| import sys | |
| from loopcut import cut_loops | |
| with open(sys.argv[2], "w") as f: | |
| for l in open(sys.argv[1]): | |
| r = json.loads(l) | |
| raw = r.get("pred_raw", r["pred"]) | |
| f.write(json.dumps(dict(r, pred=cut_loops(raw or ""), pred_raw=raw), ensure_ascii=False) + "\n") | |