ruotianluo/ImageCaptioning.pytorch

Health v2 · maintenance only

Flags: no_releases

Adoption not part of the score

1476 stars · 420 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

A PyTorch research codebase for image captioning, supporting self-critical sequence training, bottom-up features, transformer captioning models, multi-GPU training, and test-time ensembling. It includes pretrained models, training/evaluation tools for COCO and Flickr30k, and a demo Colab notebook.

Use cases

• train an image captioning model on COCO
• generate captions for images with pretrained models
• apply self-critical sequence training to captioning
• evaluate captioning models with BLEU/METEOR/CIDEr
• train captioning models on Flickr30k
• run multi-GPU distributed training for captioning

When to choose

• you need a proven research codebase for image captioning with self-critical training
• you want pretrained captioning models for COCO or Flickr30k
• you need bottom-up attention features or transformer captioning models in PyTorch

When to avoid

• you need production-ready image captioning as a service
• you want a general-purpose vision-language framework beyond captioning
• you need active development or frequent updates

Facets

library · maturity maintenance

machine-learning deep-learning nlp image-processing machine-learning deep-learning computer-vision python image-captioning pytorch self-critical-training coco transformer research-codebase natural-language-processing gpu

Member repositories

For agents

markdown · JSON · MCP: product_card(name="ruotianluo/ImageCaptioning.pytorch")