Skip to content

Commit f8a7d45

Browse files
committed
Bug fix for label_map for publaynet
1 parent 2355663 commit f8a7d45

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

docs/notes/modelzoo.md

Lines changed: 1 addition & 1 deletion
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) | `{0:"Text", 1:"Title", 2:"List", 3:"Table", 4:"Figure"}` |
33+
| [PubLayNet](https://github.com/ibm-aur-nlp/PubLayNet) | `{1: "Text", 2: "Title", 3: "List", 4:"Table", 5:"Figure"}` |
3434
| [PrimaLayout](https://www.primaresearch.org/dataset/) | `{1:"TextRegion", 2:"ImageRegion", 3:"TableRegion", 4:"MathsRegion", 5:"SeparatorRegion", 6:"OtherRegion"}` |

examples/Deep Layout Parsing.ipynb

Lines changed: 1 addition & 1 deletion
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={0: \"Text\", 1: \"Title\", 2: \"List\", 3:\"Table\", 4:\"Figure\"})\n",
61+
" label_map={1: \"Text\", 2: \"Title\", 3: \"List\", 4:\"Table\", 5:\"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"

0 commit comments

Comments
 (0)