| import layoutparser as lp | |
| import cv2 | |
| image = cv2.imread("data/img.png") | |
| image = image[..., ::-1] | |
| # Convert the image from BGR (cv2 default loading style) | |
| # to RGB | |
| model = lp.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 | |
| # For all the supported model, please check the Model | |
| # Zoo Page: https://layout-parser.readthedocs.io/en/latest/notes/modelzoo.html | |
| layout = model.detect(image) | |
| # Detect the layout of the input image | |
| lp.draw_box(image, layout, box_width=3) | |
| # Show the detected layout of the input image |