소스 검색

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,