fixes for paddle ocr

2024-10-31 15:18:12 -07:00
parent 612785d1c0
commit b094079a06
7 changed files with 117 additions and 138 deletions
--- a/.gitignore
+++ b/.gitignore
@@ -2,4 +2,4 @@ weights/icon_caption_blip2
 weights/icon_caption_florence
 weights/icon_detect/
 .gradio
-__pycache__
+__pycache__/
--- a/pycache/utils.cpython-312.pyc
+++ b/pycache/utils.cpython-312.pyc
--- a/pycache/utils.cpython-39.pyc
+++ b/pycache/utils.cpython-39.pyc
--- a/demo.ipynb
+++ b/demo.ipynb
--- a/gradio_demo.py
+++ b/gradio_demo.py
@@ -90,7 +90,7 @@ with gr.Blocks() as demo:
            iou_threshold_component = gr.Slider(
                label='IOU Threshold', minimum=0.01, maximum=1.0, step=0.01, value=0.1)
            use_paddleocr_component = gr.Checkbox(
-                label='Use PaddleOCR', default=True)
+                label='Use PaddleOCR', value=True)
            submit_button_component = gr.Button(
                value='Submit', variant='primary')
        with gr.Column():
--- a/imgs/saved_image_demo.png
+++ b/imgs/saved_image_demo.png
--- a/utils.py
+++ b/utils.py
@@ -411,7 +411,7 @@ def check_ocr_box(image_path, display_img = True, output_bb_format='xywh', goal_
        elif output_bb_format == 'xyxy':
            bb = [get_xyxy(item) for item in coord]
        # print('bounding box!!!', bb)
-    return (text, bb), is_goal_filtered
+    return (text, bb), goal_filtering