From 911a5610703a4ee2d7438e1398c5bbe8079099cc Mon Sep 17 00:00:00 2001 From: HsiangNianian Date: Mon, 5 Jan 2026 19:17:08 +0800 Subject: refactor: remove unused fix_speaker function and clean up code in onnx_infer.py --- models/trpg-final/model.onnx | Bin 776319 -> 776319 bytes models/trpg-final/model.onnx.data | Bin 41164800 -> 41164800 bytes tests/onnx_infer.py | 16 ++++------------ 3 files changed, 4 insertions(+), 12 deletions(-) diff --git a/models/trpg-final/model.onnx b/models/trpg-final/model.onnx index e5e585c..b125d02 100644 Binary files a/models/trpg-final/model.onnx and b/models/trpg-final/model.onnx differ diff --git a/models/trpg-final/model.onnx.data b/models/trpg-final/model.onnx.data index 9465ee4..fa5b018 100644 Binary files a/models/trpg-final/model.onnx.data and b/models/trpg-final/model.onnx.data differ diff --git a/tests/onnx_infer.py b/tests/onnx_infer.py index d6c00d2..31cc546 100644 --- a/tests/onnx_infer.py +++ b/tests/onnx_infer.py @@ -116,14 +116,6 @@ def fix_timestamp(ts): return ts -def fix_speaker(spk): - if not spk: - return spk - spk = re.sub(r"[^\w\s\u4e00-\u9fff]+$", "", spk) - if len(spk) == 1 and re.match(r"^[风雷电雨雪火水木金]", spk): - return spk + "某" - return spk - out = {"metadata": {}, "content": []} for e in entities: @@ -132,10 +124,10 @@ for e in entities: conf = round(float(np.mean(e["probs"])), 3) typ = e["type"] if typ in ("timestamp", "speaker"): - if typ == "timestamp": - ent_text = fix_timestamp(ent_text) - else: - ent_text = fix_speaker(ent_text) + # if typ == "timestamp": + # ent_text = fix_timestamp(ent_text) + # else: + # ent_text = ent_text out["metadata"][typ] = ent_text else: out["content"].append({"type": typ, "content": ent_text, "confidence": conf}) -- cgit v1.2.3-70-g09d2