microsoft
/

Florence-2-large

Image-Text-to-Text

text-generation

Model card Files Files and versions Community

update_ocr_postprocess

#52

by haipingwu - opened Jul 18

base: refs/heads/main

←

from: refs/pr/52

Discussion Files changed

Files changed (1) hide show

processing_florence2.py +2 -2

processing_florence2.py CHANGED Viewed

@@ -596,7 +596,7 @@ class Florence2PostProcesser(object):
                 {
                     'TASK_NAME': 'ocr',
                     'PATTERN':  r'(.+?)<loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)>',
-                    'AREA_THRESHOLD': 0.01
                 },
                 {
                     'TASK_NAME': 'phrase_grounding',
@@ -1025,7 +1025,7 @@ class Florence2PostProcesser(object):
                     text,
                     pattern=pattern,
                     image_size=image_size,
-                    area_threshold=self.parse_tasks_configs[task].get('AREA_THRESHOLD', 0.01),
                 )
                 parsed_dict['ocr'] = instances
             elif task == 'phrase_grounding':

                 {
                     'TASK_NAME': 'ocr',
                     'PATTERN':  r'(.+?)<loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)>',
+                    'AREA_THRESHOLD': 0.00
                 },
                 {
                     'TASK_NAME': 'phrase_grounding',
                     text,
                     pattern=pattern,
                     image_size=image_size,
+                    area_threshold=self.parse_tasks_configs[task].get('AREA_THRESHOLD', 0.0),
                 )
                 parsed_dict['ocr'] = instances
             elif task == 'phrase_grounding':