[Help] looking for image captioning model based on transformer

tensorflow / tensor2tensor

Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.

Apache License 2.0

15.31k stars 3.47k forks source link

[Help] looking for image captioning model based on transformer #1547

Open ColorBuffer opened 5 years ago

ColorBuffer commented 5 years ago

Hi, im looking for a image captioning model based on transformer. Does it exist in tensor2tensor?

niekvanderlaan commented 5 years ago

I have the same question!

lukemelas commented 5 years ago

Just following up on this. It's definitely there (MSCoco is available as a dataset after all) and it was used for Conceptual Captions baselines. If anyone has an example training script though (for any dataset), that would be awesome.