Arthur commited on
Commit ·
3373f2c
1
Parent(s): a30aa26
Fix : OCR_SIGNATURES
Browse files- processing_florence2.py +1 -4
processing_florence2.py
CHANGED
|
@@ -120,7 +120,7 @@ class Florence2Processor(ProcessorMixin):
|
|
| 120 |
'<OD>': 'Locate the objects with category name in the image.',
|
| 121 |
'<DENSE_REGION_CAPTION>': 'Locate the objects in the image, with their descriptions.',
|
| 122 |
'<REGION_PROPOSAL>': 'Locate the region proposals in the image.',
|
| 123 |
-
'<OCR_SIGNATURES>': '
|
| 124 |
}
|
| 125 |
|
| 126 |
self.task_prompts_with_input = {
|
|
@@ -622,9 +622,6 @@ class Florence2PostProcesser(object):
|
|
| 622 |
},
|
| 623 |
{
|
| 624 |
'TASK_NAME': 'description_with_bboxes_or_polygons',
|
| 625 |
-
},
|
| 626 |
-
{
|
| 627 |
-
'TASK_NAME': 'ocr_signatures',
|
| 628 |
}
|
| 629 |
]
|
| 630 |
}
|
|
|
|
| 120 |
'<OD>': 'Locate the objects with category name in the image.',
|
| 121 |
'<DENSE_REGION_CAPTION>': 'Locate the objects in the image, with their descriptions.',
|
| 122 |
'<REGION_PROPOSAL>': 'Locate the region proposals in the image.',
|
| 123 |
+
'<OCR_SIGNATURES>': 'Find the signatures in the image if any and return them'
|
| 124 |
}
|
| 125 |
|
| 126 |
self.task_prompts_with_input = {
|
|
|
|
| 622 |
},
|
| 623 |
{
|
| 624 |
'TASK_NAME': 'description_with_bboxes_or_polygons',
|
|
|
|
|
|
|
|
|
|
| 625 |
}
|
| 626 |
]
|
| 627 |
}
|