ImageCaptioning.pytorch

Image captioning framework

A Python-based framework for training and testing image captioning models using PyTorch.

I decide to sync up this repo and self-critical.pytorch. (The old master is in old master branch for archive)

GitHub

1k stars
25 watching
419 forks
Language: Python
last commit: about 1 year ago
Linked from 2 awesome lists


Backlinks from these awesome lists:

Related projects:

Repository Description Stars
ruotianluo/self-critical.pytorch An implementation of Self-critical Sequence Training for Image Captioning and related techniques. 998
kacky24/stylenet A PyTorch implementation of a framework for generating captions with styles for images and videos. 63
felixgwu/img_classification_pk_pytorch A PyTorch project for comparing image classification models and facilitating quick experiment setup 366
xiadingz/video-caption.pytorch PyTorch implementation of video captioning, combining deep learning and computer vision techniques. 402
fartashf/vsepp A PyTorch implementation of visual-semantic embedding methods for image-caption retrieval 492
kaiyangzhou/dassl.pytorch A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. 1,236
mrzhu-cool/pix2pix-pytorch An implementation of an image-to-image translation algorithm using deep learning and PyTorch 428
lucasjinreal/pytorch_image_classifier An image classification system built using PyTorch, capable of fine-tuning on various models and datasets. 113
nickjiang2378/vl-interp This project provides an official PyTorch implementation of a method to interpret and edit vision-language representations to mitigate hallucinations in image captions. 46
blade6570/photographicimagesynthesiswithcascadedrefinementnetworks-pytorch An implementation of cascaded refinement networks for generating photorealistic images from semantic layouts 65
tengshaofeng/residualattentionnetwork-pytorch Implementing a deep learning framework for image classification using Residual Attention Network architecture 682
bearpaw/pytorch-classification An open-source project providing a unified interface for training and evaluating classification models using PyTorch. 1,699
chingyaoc/pytorch-reinforce A PyTorch implementation of the REINFORCE algorithm for reinforcement learning in continuous and discrete environments. 266
anuragranj/back2future.pytorch An implementation of unsupervised learning for multi-frame optical flow with occlusions using PyTorch. 112
4uiiurz1/pytorch-res2net Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. 112