From 6c11df1eebbed5a9f2136301a85c481cf663f047 Mon Sep 17 00:00:00 2001
From: Zhihao Liu <35885077+SkyFlap@users.noreply.github.com>
Date: Thu, 24 Oct 2024 19:39:35 +0800
Subject: [PATCH] Update model_init.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

修复在内部二次调用时会出现`AssertionError: Dataset 'scihub_train' is already registered!`的报错问题
---
 .../models/layoutlmv3_util/model_init.py           | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

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