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 |
|