Explorar el Código

refactor(paddle_to_pytorch): update model list in batch conversion function

zhch158_admin hace 3 semanas
padre
commit
982a9f7349
Se han modificado 1 ficheros con 2 adiciones y 1 borrados
  1. 2 1
      ocr_tools/pytorch_models/paddle_to_pytorch_universal.py

+ 2 - 1
ocr_tools/pytorch_models/paddle_to_pytorch_universal.py

@@ -168,7 +168,8 @@ def batch_convert_all_models():
         # ("PicoDet_layout_1x", "Layout"),
         # ("PP-LCNet_x1_0_doc_ori", "Layout"),
         # ("RT-DETR-H_layout_17cls", "Layout"),
-        ("PP-DocLayout_plus-L", "Layout"),
+        # ("PP-DocLayout_plus-L", "Layout"),
+        ("RT-DETR-L_wired_table_cell_det", "Table"),
     ]
     
     base_dir = Path("~/.paddlex/official_models").expanduser()