Shypd.aiThe largest open index of AI toolsself-critical.pytorch is an open-source PyTorch implementation for image captioning research. It supports self-critical sequence training, bottom-up feature extraction, and multi-GPU training.
Trending
About
self-critical.pytorch provides a comprehensive codebase for image captioning research, offering an unofficial PyTorch implementation for Self-critical Sequence Training. Key features include support for bottom-up features, test-time ensemble, and multi-GPU training, with DistributedDataParallel now supported via pytorch-lightning. The codebase also integrates Transformer captioning models and offers a simple demo via a Colab notebook. Researchers can train networks on datasets like COCO and Flickr30k, with options for scheduled sampling and evaluation using metrics like BLEU, METEOR, and CIDEr. Pretrained models are available, and the tool facilitates generating image captions and evaluating them on various splits.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
Also listed in
