From 2cb9db435f7fb99a1e79300baf8e7ed2df3cdbea Mon Sep 17 00:00:00 2001
From: lolipopshock <22512825+lolipopshock@users.noreply.github.com>
Date: Thu, 16 Jul 2020 16:34:21 -0400
Subject: [PATCH] Update Model testing file

---
 tests/test_model.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/tests/test_model.py b/tests/test_model.py
index b6dbf43..90e42d1 100644
--- a/tests/test_model.py
+++ b/tests/test_model.py
@@ -16,11 +16,7 @@ def test_Detectron2Model(is_large_scale=False):
     if is_large_scale:
         
         for config in ALL_CONFIGS:  
-            model = Detectron2LayoutModel(
-                config,
-                extra_config=["MODEL.ROI_HEADS.SCORE_THRESH_TEST", 0.8],
-                label_map={0: "Text", 1: "Title", 2: "List", 3:"Table", 4:"Figure"}
-            )
+            model = Detectron2LayoutModel(config)
             
             image = cv2.imread("tests/source/test_gcv_image.jpg")
             layout = model.detect(image)