Skip to content

Commit 2cb9db4

Browse files
committed
Update Model testing file
1 parent 58ce8e4 commit 2cb9db4

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

tests/test_model.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,7 @@ def test_Detectron2Model(is_large_scale=False):
1616
if is_large_scale:
1717

1818
for config in ALL_CONFIGS:
19-
model = Detectron2LayoutModel(
20-
config,
21-
extra_config=["MODEL.ROI_HEADS.SCORE_THRESH_TEST", 0.8],
22-
label_map={0: "Text", 1: "Title", 2: "List", 3:"Table", 4:"Figure"}
23-
)
19+
model = Detectron2LayoutModel(config)
2420

2521
image = cv2.imread("tests/source/test_gcv_image.jpg")
2622
layout = model.detect(image)

0 commit comments

Comments
 (0)