From 33af39ed1ec536db7479ae31b2ad814c10bf7feb Mon Sep 17 00:00:00 2001 From: drewm23 <43488758+drewm23@users.noreply.github.com> Date: Mon, 18 Sep 2023 17:26:53 -0700 Subject: [PATCH] Update index.rst allow for proper loading of Detectron2LayoutModel --- docs/example/deep_layout_parsing/index.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/example/deep_layout_parsing/index.rst b/docs/example/deep_layout_parsing/index.rst index da0ac1a..6bdca76 100644 --- a/docs/example/deep_layout_parsing/index.rst +++ b/docs/example/deep_layout_parsing/index.rst @@ -29,7 +29,7 @@ Use Layout Models to detect complex layout .. code:: python - model = lp.Detectron2LayoutModel('lp://PubLayNet/faster_rcnn_R_50_FPN_3x/config', + model = lp.models.Detectron2LayoutModel('lp://PubLayNet/faster_rcnn_R_50_FPN_3x/config', extra_config=["MODEL.ROI_HEADS.SCORE_THRESH_TEST", 0.8], label_map={0: "Text", 1: "Title", 2: "List", 3:"Table", 4:"Figure"}) # Load the deep layout model from the layoutparser API