Image-to-Text
Transformers
Safetensors
vision-encoder-decoder
image-text-to-text
Generated from Trainer
Instructions to use mo-thecreator/ViT-GPT2-Image-Captioning with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use mo-thecreator/ViT-GPT2-Image-Captioning with Transformers:
# Use a pipeline as a high-level helper # Warning: Pipeline type "image-to-text" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # 'pip install "transformers<5.0.0' from transformers import pipeline pipe = pipeline("image-to-text", model="mo-thecreator/ViT-GPT2-Image-Captioning")# Load model directly from transformers import AutoTokenizer, AutoModelForImageTextToText tokenizer = AutoTokenizer.from_pretrained("mo-thecreator/ViT-GPT2-Image-Captioning") model = AutoModelForImageTextToText.from_pretrained("mo-thecreator/ViT-GPT2-Image-Captioning") - Notebooks
- Google Colab
- Kaggle
Mohammed Abdeldayem commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -18,11 +18,11 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 18 |
|
| 19 |
This model is a fine-tuned version of [motheecreator/ViT-GPT2-Image_Captioning_model](https://huggingface.co/motheecreator/ViT-GPT2-Image_Captioning_model) on an unknown dataset.
|
| 20 |
It achieves the following results on the evaluation set:
|
| 21 |
-
- Loss: 2.
|
| 22 |
- Rouge2 Precision: None
|
| 23 |
- Rouge2 Recall: None
|
| 24 |
-
- Rouge2 Fmeasure: 0.
|
| 25 |
-
- Bleu: 9.
|
| 26 |
|
| 27 |
## Model description
|
| 28 |
|
|
@@ -56,8 +56,8 @@ The following hyperparameters were used during training:
|
|
| 56 |
|
| 57 |
| Training Loss | Epoch | Step | Validation Loss | Rouge2 Precision | Rouge2 Recall | Rouge2 Fmeasure | Bleu |
|
| 58 |
|:-------------:|:------:|:----:|:---------------:|:----------------:|:-------------:|:---------------:|:------:|
|
| 59 |
-
| 2.
|
| 60 |
-
| 2.
|
| 61 |
|
| 62 |
|
| 63 |
### Framework versions
|
|
|
|
| 18 |
|
| 19 |
This model is a fine-tuned version of [motheecreator/ViT-GPT2-Image_Captioning_model](https://huggingface.co/motheecreator/ViT-GPT2-Image_Captioning_model) on an unknown dataset.
|
| 20 |
It achieves the following results on the evaluation set:
|
| 21 |
+
- Loss: 2.125337
|
| 22 |
- Rouge2 Precision: None
|
| 23 |
- Rouge2 Recall: None
|
| 24 |
+
- Rouge2 Fmeasure: 0.155
|
| 25 |
+
- Bleu: 9.7054
|
| 26 |
|
| 27 |
## Model description
|
| 28 |
|
|
|
|
| 56 |
|
| 57 |
| Training Loss | Epoch | Step | Validation Loss | Rouge2 Precision | Rouge2 Recall | Rouge2 Fmeasure | Bleu |
|
| 58 |
|:-------------:|:------:|:----:|:---------------:|:----------------:|:-------------:|:---------------:|:------:|
|
| 59 |
+
| 2.1537 | 0.9993 | 1171 | 2.13666 | None | None | 0.1531 | 9.4673 |
|
| 60 |
+
| 2.0434 | 1.9985 | 2342 | 2.125337 | None | None | 0.155 | 9.7054 |
|
| 61 |
|
| 62 |
|
| 63 |
### Framework versions
|