Dom/OmniParser
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixes for paddle ocr

Browse Source
This commit is contained in:
yadonglu
2024-10-31 15:18:12 -07:00
parent 612785d1c0
commit b094079a06
7 changed files with 117 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

@@ -2,4 +2,4 @@ weights/icon_caption_blip2
weights/icon_caption_florence weights/icon_caption_florence
weights/icon_detect/ weights/icon_detect/
.gradio .gradio
__pycache__ __pycache__/

BIN
__pycache__/utils.cpython-312.pyc
View File

Binary file not shown.

BIN
__pycache__/utils.cpython-39.pyc
View File

Binary file not shown.

249
demo.ipynb
View File

File diff suppressed because one or more lines are too long

2
gradio_demo.py
View File

@@ -90,7 +90,7 @@ with gr.Blocks() as demo:
iou_threshold_component = gr.Slider( iou_threshold_component = gr.Slider(
label='IOU Threshold', minimum=0.01, maximum=1.0, step=0.01, value=0.1) label='IOU Threshold', minimum=0.01, maximum=1.0, step=0.01, value=0.1)
use_paddleocr_component = gr.Checkbox( use_paddleocr_component = gr.Checkbox(
label='Use PaddleOCR', default=True) label='Use PaddleOCR', value=True)
submit_button_component = gr.Button( submit_button_component = gr.Button(
value='Submit', variant='primary') value='Submit', variant='primary')
with gr.Column(): with gr.Column():

BIN
imgs/saved_image_demo.png
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 341 KiB

After

Width:  |  Height:  |  Size: 219 KiB

2
utils.py
View File

@@ -411,7 +411,7 @@ def check_ocr_box(image_path, display_img = True, output_bb_format='xywh', goal_
elif output_bb_format == 'xyxy': elif output_bb_format == 'xyxy':
bb = [get_xyxy(item) for item in coord] bb = [get_xyxy(item) for item in coord]
# print('bounding box!!!', bb) # print('bounding box!!!', bb)
return (text, bb), is_goal_filtered return (text, bb), goal_filtering