/image_caption

An image caption model for learning

Primary LanguagePython

image_caption

An image caption model built by ResNet34 and Transformer. The dataset for training comes from AI Challenger.

The pre-trained model can be downloaded here.