Parcourir la source

fix(models): update unimernet_small model path

- Update model path from 'unimernet_small' to 'unimernet_small_2501' in multiple scripts and configuration files
- This change affects download_models.py, download_models_hf.py, and model_configs.yaml
myhloli il y a 10 mois
Parent
commit
2a3a006f4d

+ 1 - 1
magic_pdf/resources/model_config/model_configs.yaml

@@ -2,7 +2,7 @@ weights:
   layoutlmv3: Layout/LayoutLMv3/model_final.pth
   doclayout_yolo: Layout/YOLO/doclayout_yolo_docstructbench_imgsz1280_2501.pt
   yolo_v8_mfd: MFD/YOLO/yolo_v8_ft.pt
-  unimernet_small: MFR/unimernet_small
+  unimernet_small: MFR/unimernet_small_2501
   struct_eqtable: TabRec/StructEqTable
   tablemaster: TabRec/TableMaster
   rapid_table: TabRec/RapidTable

BIN
projects/gradio_app/examples/complex_layout.pdf


+ 1 - 1
scripts/download_models.py

@@ -35,7 +35,7 @@ if __name__ == '__main__':
         "models/Layout/LayoutLMv3/*",
         "models/Layout/YOLO/*",
         "models/MFD/YOLO/*",
-        "models/MFR/unimernet_small/*",
+        "models/MFR/unimernet_small_2501/*",
         "models/TabRec/TableMaster/*",
         "models/TabRec/StructEqTable/*",
     ]

+ 1 - 1
scripts/download_models_hf.py

@@ -36,7 +36,7 @@ if __name__ == '__main__':
         "models/Layout/LayoutLMv3/*",
         "models/Layout/YOLO/*",
         "models/MFD/YOLO/*",
-        "models/MFR/unimernet_small/*",
+        "models/MFR/unimernet_small_2501/*",
         "models/TabRec/TableMaster/*",
         "models/TabRec/StructEqTable/*",
     ]