ImageCaptioning.pytorch
by ruotianluo
I decide to sync up this repo and self-critical.pytorch. (The old master is in old master branch for archive)
AI summary
Image captioning framework
A Python-based framework for training and testing image captioning models using PyTorch.
- stars
- 1.5K
- forks
- 419
- watching
- 25
- awesome lists
- 2
Similar projects
Found by comparing what the projects do, not just their names.
Image captioning library
An implementation of Self-critical Sequence Training for Image Captioning and related techniques.
Caption generator
A PyTorch implementation of a framework for generating captions with styles for images and videos.
Image classifier
A PyTorch project for comparing image classification models and facilitating quick experiment setup
Video captioner
PyTorch implementation of video captioning, combining deep learning and computer vision techniques.
ImageCaptionRetrieval
A PyTorch implementation of visual-semantic embedding methods for image-caption retrieval
Computer Vision Toolkit
A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision.
Image translator
An implementation of an image-to-image translation algorithm using deep learning and PyTorch
Image classifier
An image classification system built using PyTorch, capable of fine-tuning on various models and datasets.
Hallucination mitigation
This project provides an official PyTorch implementation of a method to interpret and edit vision-language representations to mitigate hallucinations in image captions.
Image synthesis library
An implementation of cascaded refinement networks for generating photorealistic images from semantic layouts
Image classifier
Implementing a deep learning framework for image classification using Residual Attention Network architecture
Classification framework
An open-source project providing a unified interface for training and evaluating classification models using PyTorch.
Reinforcement learning framework
A PyTorch implementation of the REINFORCE algorithm for reinforcement learning in continuous and discrete environments.
Optical flow model
An implementation of unsupervised learning for multi-frame optical flow with occlusions using PyTorch.
Deep learning models
Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets.
Featured in 2 awesome lists
Each link jumps to the spot where the list mentions ImageCaptioning.pytorch.
