Update README.md
Browse files
README.md
CHANGED
|
@@ -10,6 +10,26 @@ tags:
|
|
| 10 |
- adapter
|
| 11 |
- image-captioning
|
| 12 |
- peft
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
---
|
| 14 |
|
| 15 |
# Florence-2 Recap-DataComp LoRA Adapter
|
|
@@ -57,4 +77,18 @@ This code demonstrates how to:
|
|
| 57 |
2. Load the LoRA adapter
|
| 58 |
3. Process an image and generate a detailed caption
|
| 59 |
|
| 60 |
-
Note: Make sure you have the required libraries installed: transformers, peft, einops, flash_attn, timm, Pillow, and requests.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
- adapter
|
| 11 |
- image-captioning
|
| 12 |
- peft
|
| 13 |
+
model-index:
|
| 14 |
+
- name: Florence-2-DOCCI-FT
|
| 15 |
+
results:
|
| 16 |
+
- task:
|
| 17 |
+
type: image-to-text
|
| 18 |
+
name: Image Captioning
|
| 19 |
+
dataset:
|
| 20 |
+
name: foundation-multimodal-models/DetailCaps-4870
|
| 21 |
+
type: other
|
| 22 |
+
metrics:
|
| 23 |
+
- type: meteor
|
| 24 |
+
value: 0.240
|
| 25 |
+
- type: bleu
|
| 26 |
+
value: 0.150
|
| 27 |
+
- type: cider
|
| 28 |
+
value: 0.035
|
| 29 |
+
- type: capture
|
| 30 |
+
value: 0.553
|
| 31 |
+
- type: rouge-l
|
| 32 |
+
value: 0.294
|
| 33 |
---
|
| 34 |
|
| 35 |
# Florence-2 Recap-DataComp LoRA Adapter
|
|
|
|
| 77 |
2. Load the LoRA adapter
|
| 78 |
3. Process an image and generate a detailed caption
|
| 79 |
|
| 80 |
+
Note: Make sure you have the required libraries installed: transformers, peft, einops, flash_attn, timm, Pillow, and requests.
|
| 81 |
+
|
| 82 |
+
## Evaluation results
|
| 83 |
+
|
| 84 |
+
Our LoRA adapter shows improvements over the base Florence-2 model across all metrics for MORE_DETAILED_CAPTION tag for 1000 images on the foundation-multimodal-models/DetailCaps-4870 dataset:
|
| 85 |
+
|
| 86 |
+
| Metric | Base Model | Adapted Model | Improvement |
|
| 87 |
+
|---------|------------|-----------------------|-------------|
|
| 88 |
+
| METEOR | 0.213 | 0.240 | +12.7% |
|
| 89 |
+
| BLEU | 0.110 | 0.150 | +36.4% |
|
| 90 |
+
| CIDEr | 0.031 | 0.035 | +12.9% |
|
| 91 |
+
| CAPTURE | 0.546 | 0.553 | +1.3% |
|
| 92 |
+
| ROUGE-L | 0.275 | 0.294 | +6.9% |
|
| 93 |
+
|
| 94 |
+
These results demonstrate that our LoRA adapter enhances the image captioning capabilities of the Florence-2 base model, particularly in generating more detailed and accurate captions.
|