Open Aman-Preet-Singh-Gulati opened 11 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.
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'
Install dectron2 0.6 and Pillow==9.50 fixed this issue.
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
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 model
label_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)
label_map
5 extra_config=["MODEL.ROI_HEADS.SCORE_THRESH_TEST", 0.7] # Optional16 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'