diff --git a/pdf_extract_kit/tasks/layout_detection/models/layoutlmv3_util/model_init.py b/pdf_extract_kit/tasks/layout_detection/models/layoutlmv3_util/model_init.py index cce7a81..554d78b 100644 --- a/pdf_extract_kit/tasks/layout_detection/models/layoutlmv3_util/model_init.py +++ b/pdf_extract_kit/tasks/layout_detection/models/layoutlmv3_util/model_init.py @@ -73,12 +73,12 @@ def setup(args): cfg.freeze() default_setup(cfg, args) - register_coco_instances( - "scihub_train", - {}, - cfg.SCIHUB_DATA_DIR_TRAIN + ".json", - cfg.SCIHUB_DATA_DIR_TRAIN - ) + # register_coco_instances( + # "scihub_train", + # {}, + # cfg.SCIHUB_DATA_DIR_TRAIN + ".json", + # cfg.SCIHUB_DATA_DIR_TRAIN + # ) return cfg @@ -138,4 +138,4 @@ def __call__(self, image, ignore_catids=[]): ], "score": scores[bbox_idx] }) - return page_layout_result \ No newline at end of file + return page_layout_result