# Image captioning for Indonesia with pre-trained vision

**URL:** </static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fimage-captioning-for-indonesia-with-pre-trained-vision%2F7285%3E%5C
**Category:** Flax/JAX Projects\
**Created:** [June 24, 2021, 6:40pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fimage-captioning-for-indonesia-with-pre-trained-vision%2F7285 "2021-06-24T18:40:23Z")\
**Posts on this page:** 5\
**Page:** 1

<div class="post-metadata">

**Author:** ![munggok](https://avatars.discourse-cdn.com/v4/letter/m/919ad9/32.png) [@munggok](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fmunggok)%5C
**Post date:** [June 24, 2021, 6:40pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fimage-captioning-for-indonesia-with-pre-trained-vision%2F7285%2F1 "2021-06-24T18:40:23Z")

</div>

## Image captioning for Indonesia with pre-trained vision and text model

For this project, a pre-trained image model like `ViT` can be used as an encoder, and a pre-trained text model like BERT and/or GPT2 can be used as a decoder.

### Model

Pre-trained [ViT](https://huggingface.co/models?filter=vit), [BERT](https://huggingface.co/bert-base-multilingual-cased) models can be found on the model hub. We could use multi-lingual BERT/ROERTa models for the Indonesian language.

### Datasets

The [WIT](https://github.com/google-research-datasets/wit) dataset can be used for this task. It has almost over 200K image-text pairs for Indonesia.  
The [GEM](https://github.com/microsoft/GEM#get-dataset) dataset can also be used for the task

### Available training scripts

As this will be a Seq2Seq model, the [run\_summarization\_flax.py](https://github.com/huggingface/transformers/tree/master/examples/flax/summarization) script can be used for training this model with some modifications.

### (Optional) Desired project outcome

The desired outcome is to see if pre-trained vision and text models can be leveraged for image captioning and also train captioning models in for the Indonesia language. This can be showcased with a streamlit or gradio app.

### (Optional) Challenges

-This model will require some modifications to the existing text models.  
-Data Processing

---

<div class="post-metadata">

**Author:** ![Galuh](https://avatars.discourse-cdn.com/v4/letter/g/b2d939/32.png) [@Galuh](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FGaluh)%5C
**Post date:** [June 24, 2021, 11:16pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fimage-captioning-for-indonesia-with-pre-trained-vision%2F7285%2F2 "2021-06-24T23:16:53Z")

</div>

Count me in!

---

<div class="post-metadata">

**Author:** ![cahya](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/cahya/32/48725_2.png) [@cahya](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fcahya)%5C
**Post date:** [June 25, 2021, 7:05pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fimage-captioning-for-indonesia-with-pre-trained-vision%2F7285%2F3 "2021-06-25T19:05:56Z")

</div>

I am also in

---

<div class="post-metadata">

**Author:** ![ayameRushia](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/ayamerushia/32/16473_2.png) [@ayameRushia](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2FayameRushia)%5C
**Post date:** [June 29, 2021, 9:28am UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fimage-captioning-for-indonesia-with-pre-trained-vision%2F7285%2F4 "2021-06-29T09:28:41Z")

</div>

I am also in

---

<div class="post-metadata">

**Author:** ![patrickvonplaten](https://sea2.discourse-cdn.com/hellohellohello/user_avatar/discuss.huggingface.co/patrickvonplaten/32/2171_2.png) [@patrickvonplaten](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Fu%2Fpatrickvonplaten)%5C
**Post date:** [June 29, 2021, 3:16pm UTC](/static-proxy?url=https%3A%2F%2Fdiscuss.huggingface.co%2Ft%2Fimage-captioning-for-indonesia-with-pre-trained-vision%2F7285%2F5 "2021-06-29T15:16:39Z")

</div>

Great! Let’s define the project 🙂 cc @valhalla
