Skip to content

Commit 3bf0ce1

Browse files
committed
Update model zoo label_map
1 parent f8a7d45 commit 3bf0ce1

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

docs/notes/modelzoo.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,5 +30,5 @@ model.detect(image)
3030
| Dataset | Label Map |
3131
| ------------------------------------------------------------ | ------------------------------------------------------------ |
3232
| [HJDataset](https://dell-research-harvard.github.io/HJDataset/) | `{1:"Page Frame", 2:"Row", 3:"Title Region", 4:"Text Region", 5:"Title", 6:"Subtitle", 7:"Other"}` |
33-
| [PubLayNet](https://github.com/ibm-aur-nlp/PubLayNet) | `{1: "Text", 2: "Title", 3: "List", 4:"Table", 5:"Figure"}` |
34-
| [PrimaLayout](https://www.primaresearch.org/dataset/) | `{1:"TextRegion", 2:"ImageRegion", 3:"TableRegion", 4:"MathsRegion", 5:"SeparatorRegion", 6:"OtherRegion"}` |
33+
| [PubLayNet](https://github.com/ibm-aur-nlp/PubLayNet) | `{0: "Text", 1: "Title", 2: "List", 3:"Table", 4:"Figure"}` |
34+
| [PrimaLayout](https://www.primaresearch.org/dataset/) | `{0:"TextRegion", 1:"ImageRegion", 2:"TableRegion", 3:"MathsRegion", 4:"SeparatorRegion", 5:"OtherRegion"}` |

examples/Deep Layout Parsing.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@
5858
"source": [
5959
"model = lp.Detectron2LayoutModel('lp://PubLayNet/faster_rcnn_R_50_FPN_3x/config', \n",
6060
" extra_config=[\"MODEL.ROI_HEADS.SCORE_THRESH_TEST\", 0.8],\n",
61-
" label_map={1: \"Text\", 2: \"Title\", 3: \"List\", 4:\"Table\", 5:\"Figure\"})\n",
61+
" label_map={0: \"Text\", 1: \"Title\", 2: \"List\", 3:\"Table\", 4:\"Figure\"})\n",
6262
" # Load the deep layout model from the layoutparser API \n",
6363
" # For all the supported model, please check the Model \n",
6464
" # Zoo Page: https://layout-parser.readthedocs.io/en/latest/notes/modelzoo.html"
@@ -392,4 +392,4 @@
392392
},
393393
"nbformat": 4,
394394
"nbformat_minor": 4
395-
}
395+
}

0 commit comments

Comments
 (0)