瀏覽代碼

fix post process in unimernet (#4213)

liuhongen1234567 5 月之前
父節點
當前提交
718c3a9c6c
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/inference/models/formula_recognition/processors.py

+ 1 - 1
paddlex/inference/models/formula_recognition/processors.py

@@ -798,7 +798,7 @@ class UniMERNetDecode(object):
             for i in reversed(range(len(toks[b]))):
                 if toks[b][i] is None:
                     toks[b][i] = ""
-                toks[b][i] = toks[b][i].replace("", " ").strip()
+                toks[b][i] = toks[b][i].replace("Ġ", " ").strip()
                 if toks[b][i] in (
                     [
                         self.tokenizer.bos_token,