refactor(layoutlmv3): remove outdated COCO instances registration

724001df · myhloli · 13788ca1 · 724001df
Commit 724001df authored Jul 15, 2024 by myhloli
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 6 deletions

model_init.py magic_pdf/model/pek_sub_modules/layoutlmv3/model_init.py +7 -6

No files found.
--- a/magic_pdf/model/pek_sub_modules/layoutlmv3/model_init.py
+++ b/magic_pdf/model/pek_sub_modules/layoutlmv3/model_init.py
@@ -79,12 +79,13 @@ def setup(args, device):
    cfg.freeze()
    default_setup(cfg, args)
-    register_coco_instances(
+    #@todo 可以删掉这块？
-        "scihub_train",
+    # register_coco_instances(
-        {},
+    #     "scihub_train",
-        cfg.SCIHUB_DATA_DIR_TRAIN + ".json",
+    #     {},
-        cfg.SCIHUB_DATA_DIR_TRAIN
+    #     cfg.SCIHUB_DATA_DIR_TRAIN + ".json",
-    )
+    #     cfg.SCIHUB_DATA_DIR_TRAIN
+    # )
    return cfg