python3
virtualenv -p python3 .env
source .env/bin/activate
pip install -r requirements.txt
- Automated image captioning using CNN and RNN
data_loader.py
: Load jpg images and transforms to torch Tensors (creates a batch of data and labels and pads sentences)net.py
: Network, loss function and metricstrain.py
: Train moduleevaluate.py
: Evaluate the trained model
- Show and Tell: A Neural Image Caption Generator
- Show, Attend and Tell: Neural Image Caption Generation with Visual Attention
- Where to put the Image in an Image Caption Generator
- What is the Role of Recurrent Neural Networks (RNNs) in an Image Caption Generator?
- Automated Image Captioning with ConvNets and Recurrent Nets
- Photo Caption Generator