Image captioning in icelandic
Collection
Vision model and data for image captioning in Icelandic
•
2 items
•
Updated
Author: Sigurdur Haukur Birgisson
This model is a fine-tuned version of microsoft/git-base on Sigurdur/isl-image-captioning. It achieves the following results on the evaluation set:
It appears that the model heavilly overfitted to the dataset. Also, something I failed to consider was that the base model can't write any Icelandic characters and was thus not suited for this task. Future works might want to add the capability of writing Icelandic characters to the model.
More information needed
Image captioning in Icelandic
Scraped images and their descriptions/captions from the Icelandic wikipedia.
The following hyperparameters were used during training:
Epoch | Training Loss | Validation Loss | Wer Score |
---|---|---|---|
1 | 10.096300 | 8.690205 | 102.247536 |
2 | 8.268200 | 7.655295 | 97.659365 |
3 | 7.298000 | 6.679112 | 95.714129 |
4 | 6.319800 | 5.673368 | 2.136911 |
5 | 5.317500 | 4.656871 | 22.439211 |
6 | 4.315600 | 3.667494 | 1.001095 |
7 | 3.340000 | 2.722741 | 1.063527 |
8 | 2.417700 | 1.852253 | 0.944140 |
9 | 1.593900 | 1.136962 | 0.949617 |
10 | 0.944900 | 0.638581 | 0.933187 |
11 | 0.516200 | 0.355187 | 0.955093 |
12 | 0.281600 | 0.215951 | 0.822563 |
13 | 0.167500 | 0.148763 | 0.773275 |
14 | 0.111700 | 0.116783 | 0.792990 |
15 | 0.080800 | 0.098261 | 0.729463 |
Base model
microsoft/git-base