Browse Source

add PP-DocLayoutV2 in official models (#4612)

changdazhou 1 month ago
parent
commit
a183f81c71
1 changed files with 3 additions and 2 deletions
  1. 3 2
      paddlex/inference/utils/official_models.py

+ 3 - 2
paddlex/inference/utils/official_models.py

@@ -321,7 +321,7 @@ ALL_MODELS = [
     "te_PP-OCRv5_mobile_rec",
     "ta_PP-OCRv5_mobile_rec",
     "devanagari_PP-OCRv5_mobile_rec",
-    "cyrillic_PP-OCRv5_mobile_rec"
+    "cyrillic_PP-OCRv5_mobile_rec",
 ]
 
 
@@ -360,6 +360,7 @@ OCR_MODELS = [
     "PP-DocLayout-M",
     "PP-DocLayout_plus-L",
     "PP-DocLayout-S",
+    "PP-DocLayoutV2",
     "PP-FormulaNet-L",
     "PP-FormulaNet_plus-L",
     "PP-FormulaNet_plus-M",
@@ -399,7 +400,7 @@ OCR_MODELS = [
     "te_PP-OCRv5_mobile_rec",
     "ta_PP-OCRv5_mobile_rec",
     "devanagari_PP-OCRv5_mobile_rec",
-    "cyrillic_PP-OCRv5_mobile_rec"
+    "cyrillic_PP-OCRv5_mobile_rec",
 ]