Layout-Parser / layout-parser

A Unified Toolkit for Deep Learning Based Document Image Analysis
https://layout-parser.github.io/
Apache License 2.0
4.64k stars 449 forks source link

AttributeError: module 'PIL.Image' has no attribute 'LINEAR' #187

Open Aman-Preet-Singh-Gulati opened 11 months ago

Aman-Preet-Singh-Gulati commented 11 months ago

Code: import layoutparser as lp model = lp.Detectron2LayoutModel( config_path ='lp://MFD/faster_rcnn_R_50_FPN_3x/config', # In model catalog label_map ={0: "Text", 1: "Title", 2: "List", 3:"Table", 4:"Figure"}, # In modellabel_map extra_config=["MODEL.ROI_HEADS.SCORE_THRESH_TEST", 0.7] # Optional ) layout = model.detect(image) # You need to load the image somewhere else, e.g., image = cv2.imread(...) lp.draw_box(image, layout, box_width=5, show_element_type=True)

Error:

AttributeError Traceback (most recent call last)

in <cell line: 2>() 1 import layoutparser as lp ----> 2 model = lp.Detectron2LayoutModel( 3 config_path ='lp://MFD/faster_rcnn_R_50_FPN_3x/config', # In model catalog 4 label_map ={0: "Text", 1: "Title", 2: "List", 3:"Table", 4:"Figure"}, # In modellabel_map 5 extra_config=["MODEL.ROI_HEADS.SCORE_THRESH_TEST", 0.7] # Optional

16 frames

/usr/local/lib/python3.10/dist-packages/detectron2/data/transforms/transform.py in ExtentTransform() 44 """ 45 ---> 46 def init(self, src_rect, output_size, interp=Image.LINEAR, fill=0): 47 """ 48 Args:

AttributeError: module 'PIL.Image' has no attribute 'LINEAR'

risk-seeking commented 10 months ago

I believe if you pip install Pillow==9.5.0, that can address the issue, although now I'm running into "AssertionError: Checkpoint .../model_final.pth not found!", so that's fun.

Rabbitzzc commented 9 months ago

refer: https://github.com/facebookresearch/detectron2/issues/5010#issuecomment-1649088035 fixed by python3 -m pip install -U 'git+https://github.com/facebookresearch/detectron2.git@ff53992b1985b63bd3262b5a36167098e3dada02'

yudataguy commented 9 months ago

Install dectron2 0.6 and Pillow==9.50 fixed this issue.

sdy623 commented 9 months ago

In the Pillow's 10.0 version the PIL.Image.LINEAR is deprecated and will be removed in Pillow 10, you should change it with Resampling.BILINEAR