diff --git a/docs/notes/modelzoo.md b/docs/notes/modelzoo.md index 3c7131d..aa30c97 100644 --- a/docs/notes/modelzoo.md +++ b/docs/notes/modelzoo.md @@ -6,7 +6,7 @@ We provide a spectrum of pre-trained models on different datasets. ```python import layoutparser as lp -model = lp.Detectron2LayoutModel( +model = lp.models.Detectron2LayoutModel( config_path ='lp://PubLayNet/faster_rcnn_R_50_FPN_3x/config', # In model catalog label_map ={0: "Text", 1: "Title", 2: "List", 3:"Table", 4:"Figure"}, # In model`label_map` extra_config=["MODEL.ROI_HEADS.SCORE_THRESH_TEST", 0.8] # Optional @@ -42,4 +42,4 @@ model.detect(image) | [PrimaLayout](https://www.primaresearch.org/dataset/) | `{1:"TextRegion", 2:"ImageRegion", 3:"TableRegion", 4:"MathsRegion", 5:"SeparatorRegion", 6:"OtherRegion"}` | | [NewspaperNavigator](https://news-navigator.labs.loc.gov/) | `{0: "Photograph", 1: "Illustration", 2: "Map", 3: "Comics/Cartoon", 4: "Editorial Cartoon", 5: "Headline", 6: "Advertisement"}` | | [TableBank](https://doc-analysis.github.io/tablebank-page/index.html) | `{0: "Table"}` | -| [MFD](http://transcriptorium.eu/~htrcontest/MathsICDAR2021/) | `{1: "Equation"}` | \ No newline at end of file +| [MFD](http://transcriptorium.eu/~htrcontest/MathsICDAR2021/) | `{1: "Equation"}` |