Fix Tasks examples
#4
by
Mikael110
- opened
README.md
CHANGED
|
@@ -99,7 +99,7 @@ def run_example(prompt, text_input=None):
|
|
| 99 |
)
|
| 100 |
generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
|
| 101 |
|
| 102 |
-
parsed_answer = processor.post_process_generation(generated_text, task=
|
| 103 |
|
| 104 |
print(parsed_answer)
|
| 105 |
```
|
|
@@ -113,7 +113,7 @@ Here are the tasks `Florence-2` could perform:
|
|
| 113 |
### OCR
|
| 114 |
|
| 115 |
```python
|
| 116 |
-
prompt = <OCR>
|
| 117 |
run_example(prompt)
|
| 118 |
```
|
| 119 |
|
|
@@ -121,25 +121,25 @@ run_example(prompt)
|
|
| 121 |
OCR with region output format:
|
| 122 |
{'\<OCR_WITH_REGION>': {'quad_boxes': [[x1, y1, x2, y2, x3, y3, x4, y4], ...], 'labels': ['text1', ...]}}
|
| 123 |
```python
|
| 124 |
-
prompt = <OCR_WITH_REGION>
|
| 125 |
run_example(prompt)
|
| 126 |
```
|
| 127 |
|
| 128 |
### Caption
|
| 129 |
```python
|
| 130 |
-
prompt = <CAPTION>
|
| 131 |
run_example(prompt)
|
| 132 |
```
|
| 133 |
|
| 134 |
### Detailed Caption
|
| 135 |
```python
|
| 136 |
-
prompt = <DETAILED_CAPTION>
|
| 137 |
run_example(prompt)
|
| 138 |
```
|
| 139 |
|
| 140 |
### More Detailed Caption
|
| 141 |
```python
|
| 142 |
-
prompt = <MORE_DETAILED_CAPTION>
|
| 143 |
run_example(prompt)
|
| 144 |
```
|
| 145 |
|
|
@@ -150,7 +150,7 @@ OD results format:
|
|
| 150 |
'labels': ['label1', 'label2', ...]} }
|
| 151 |
|
| 152 |
```python
|
| 153 |
-
prompt = <OD>
|
| 154 |
run_example(prompt)
|
| 155 |
```
|
| 156 |
|
|
@@ -159,7 +159,7 @@ Dense region caption results format:
|
|
| 159 |
{'\<DENSE_REGION_CAPTION>' : {'bboxes': [[x1, y1, x2, y2], ...],
|
| 160 |
'labels': ['label1', 'label2', ...]} }
|
| 161 |
```python
|
| 162 |
-
prompt = <DENSE_REGION_CAPTION>
|
| 163 |
run_example(prompt)
|
| 164 |
```
|
| 165 |
|
|
@@ -168,7 +168,7 @@ Dense region caption results format:
|
|
| 168 |
{'\<REGION_PROPOSAL>': {'bboxes': [[x1, y1, x2, y2], ...],
|
| 169 |
'labels': ['', '', ...]}}
|
| 170 |
```python
|
| 171 |
-
prompt = <REGION_PROPOSAL>
|
| 172 |
run_example(prompt)
|
| 173 |
```
|
| 174 |
|
|
|
|
| 99 |
)
|
| 100 |
generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
|
| 101 |
|
| 102 |
+
parsed_answer = processor.post_process_generation(generated_text, task=prompt, image_size=(image.width, image.height))
|
| 103 |
|
| 104 |
print(parsed_answer)
|
| 105 |
```
|
|
|
|
| 113 |
### OCR
|
| 114 |
|
| 115 |
```python
|
| 116 |
+
prompt = '<OCR>'
|
| 117 |
run_example(prompt)
|
| 118 |
```
|
| 119 |
|
|
|
|
| 121 |
OCR with region output format:
|
| 122 |
{'\<OCR_WITH_REGION>': {'quad_boxes': [[x1, y1, x2, y2, x3, y3, x4, y4], ...], 'labels': ['text1', ...]}}
|
| 123 |
```python
|
| 124 |
+
prompt = '<OCR_WITH_REGION>'
|
| 125 |
run_example(prompt)
|
| 126 |
```
|
| 127 |
|
| 128 |
### Caption
|
| 129 |
```python
|
| 130 |
+
prompt = '<CAPTION>'
|
| 131 |
run_example(prompt)
|
| 132 |
```
|
| 133 |
|
| 134 |
### Detailed Caption
|
| 135 |
```python
|
| 136 |
+
prompt = '<DETAILED_CAPTION>'
|
| 137 |
run_example(prompt)
|
| 138 |
```
|
| 139 |
|
| 140 |
### More Detailed Caption
|
| 141 |
```python
|
| 142 |
+
prompt = '<MORE_DETAILED_CAPTION>'
|
| 143 |
run_example(prompt)
|
| 144 |
```
|
| 145 |
|
|
|
|
| 150 |
'labels': ['label1', 'label2', ...]} }
|
| 151 |
|
| 152 |
```python
|
| 153 |
+
prompt = '<OD>'
|
| 154 |
run_example(prompt)
|
| 155 |
```
|
| 156 |
|
|
|
|
| 159 |
{'\<DENSE_REGION_CAPTION>' : {'bboxes': [[x1, y1, x2, y2], ...],
|
| 160 |
'labels': ['label1', 'label2', ...]} }
|
| 161 |
```python
|
| 162 |
+
prompt = '<DENSE_REGION_CAPTION>'
|
| 163 |
run_example(prompt)
|
| 164 |
```
|
| 165 |
|
|
|
|
| 168 |
{'\<REGION_PROPOSAL>': {'bboxes': [[x1, y1, x2, y2], ...],
|
| 169 |
'labels': ['', '', ...]}}
|
| 170 |
```python
|
| 171 |
+
prompt = '<REGION_PROPOSAL>'
|
| 172 |
run_example(prompt)
|
| 173 |
```
|
| 174 |
|