Skip to content

Commit 5561b92

Browse files
committed
add ocrv6 models
1 parent 6633372 commit 5561b92

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

paddlex/inference/utils/official_models.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -352,6 +352,12 @@
352352
"G2PWModel",
353353
"fastspeech2_csmsc",
354354
"pwgan_csmsc",
355+
"PP-OCRv6_medium_det",
356+
"PP-OCRv6_small_det",
357+
"PP-OCRv6_tiny_det",
358+
"PP-OCRv6_medium_rec",
359+
"PP-OCRv6_small_rec",
360+
"PP-OCRv6_tiny_rec",
355361
]
356362

357363

@@ -435,6 +441,12 @@
435441
"ta_PP-OCRv5_mobile_rec",
436442
"devanagari_PP-OCRv5_mobile_rec",
437443
"cyrillic_PP-OCRv5_mobile_rec",
444+
"PP-OCRv6_medium_det",
445+
"PP-OCRv6_small_det",
446+
"PP-OCRv6_tiny_det",
447+
"PP-OCRv6_medium_rec",
448+
"PP-OCRv6_small_rec",
449+
"PP-OCRv6_tiny_rec",
438450
]
439451

440452
SAFETENSORS_SUPPORTED_MODELS_WITH_SUFFIX: Set[str] = {

0 commit comments

Comments
 (0)