Commit52ce56b

authored

[fix] set label_map in Detectron2LayoutModel (#75)

* remove unused type annotations* automatically set label_map in Detectron2LayoutModel

1 parent73e3015 commit52ce56bCopy full SHA for 52ce56b

File tree

-3

lines changed

-3

lines changed

Lines changed: 9 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -90,6 +90,13 @@ def __init__(`
`90`	`90`	`)`
`91`	`91`	`config_path=PathManager.get_local_path(config_path)`
`92`	`92`
	`93`	`+iflabel_mapisNone:`
	`94`	`+ifconfig_path.startswith("lp://"):`
	`95`	`+dataset_name=config_path.lstrip("lp://").split("/")[1]`
	`96`	`+label_map=LABEL_MAP_CATALOG[dataset_name]`
	`97`	`+else:`
	`98`	`+label_map= {}`
	`99`	`+`
`93`	`100`	`cfg=detectron2.config.get_cfg()`
`94`	`101`	`cfg.merge_from_file(config_path)`
`95`	`102`	`cfg.merge_from_list(extra_config)`
`@@ -126,8 +133,8 @@ def gather_output(self, outputs):`
`126`	`133`	`forscore,box,labelinzip(scores,boxes,labels):`
`127`	`134`	`x_1,y_1,x_2,y_2=box`
`128`	`135`
`129`		`-ifself.label_mapisnotNone:`
`130`		`-label=self.label_map.get(label,label)`
	`136`	`+`
	`137`	`+label=self.label_map.get(label,label)`
`131`	`138`
`132`	`139`	`cur_block=TextBlock(`
`133`	`140`	`Rectangle(x_1,y_1,x_2,y_2),type=label,score=score`

Lines changed: 0 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,6 @@`
`32`	`32`	`- Brief format: lp://<dataset-name>/<config, weight>`
`33`	`33`	`"""`
`34`	`34`
`35`		`-fromtypingimportList,OrderedDict,Union,Dict,Any,Tuple,Optional,NamedTuple`
`36`	`35`	`fromdataclassesimportdataclass`
`37`	`36`
`38`	`37`	`LAYOUT_PARSER_MODEL_PREFIX="lp://"`

Comments

(0)