Skip to content
A pytorch implemention of "StyleNet: Generating Attractive Visual Captions with Styles"
Branch: master
Clone or download
Latest commit fe7dac7 Aug 14, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
sample_images make sample_images Nov 21, 2017
.gitignore add .ipynb_checkpoints Jan 15, 2018
README.md add dataset URL Aug 14, 2018
build_vocab.py cleaning with flake8 Nov 15, 2017
constant.py cleaning with flake8 Nov 15, 2017
data_loader.py create data_loader for language process Nov 21, 2017
loss.py create loss, test, train Nov 21, 2017
models.py
preprocess.py make sample_images Nov 21, 2017
sample.py create sample.py Nov 22, 2017
test.py create loss, test, train Nov 21, 2017
train.py set shuffle True Jan 15, 2018

README.md

StyleNet: Generating Attractive Visual Captions with Styles

* under development

StyleNet is a novel framework to address the task of generating attractive captions for images and videos with different styles. A novel model component, named factored LSTM is used in StyleNet, which automatically distills the style factors in the monolingual text corpus.

framework Imgur

examples of generated captions Imgur

Description

Requirement

  • python 3.5.3
  • pytorch 0.2.0
  • torchvision 0.1.9
  • numpy 1.13.3
  • scikit-image 0.13.1
  • nltk 3.2.5
You can’t perform that action at this time.