Skip to content

afshaanmaz/Caption-Generation

Repository files navigation

Caption-Generation

File Structure:

  • cococaption

  • sparsemax

     - core
    
     - data
        - annotations
             - contains json files for training and validation captions
        - test
             - contains pickle files
        - train
             - contains pickle files
        - val
             - contains pickle files
        - imagenet.mat
    
     - data_old (shows what .pkl files were available before training)
    
     - image
             - contains resized images (after preprocessing)
     - jpg
             - contains some results of attention maps
    
     - log (created after training 
             - contains tensorflow logs)
    
     - model
    
     # files
    
     - prepro.py / prepro_small.py
    
     - resize.py
    
     - test.py
    
     - train.py
    

About

Implementation of Sparse Visual Attention for Image Caption Generation. Extension of Show Attend and Tell. [Xu et. al] https://arxiv.org/abs/1502.03044

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages