Code for caption generation

[anhtienng]

Thank you for your amazing work.

Can you public the code of caption generation? I would like to evaluate on our own datasets.

[LGirrbach]

Hi @anhtienng ,

thank you very much for your interest in this work!
I'm very sorry for the slow reply.

I have added the script I used to caption the dataset, please see caption_images.py . It requires the lmdeploy library, but could be easily adapted to use vllm or a different inference framework.

Please let me know if you have any further questions.