HCBNet: Hierarchical Context-based Network

This repository is the implementation of A Hierarchical Approach for Visual Storytelling Using Image Description

Prerequisites:

1. Python 3.6
2. pytorch >= 1.0.0

Training

python train.py

Citation

@InProceedings{nahian2019hcbnet,
author="Nahian, Md. Sultan Al and Tasrin, Tasmia and Gandhi, Sagar and Gaines, Ryan and Harrison, Brent",
editor="Cardona-Rivera, Rogelio E. and Sullivan, Anne and Young, R. Michael",
title="A Hierarchical Approach for Visual Storytelling Using Image Description",
booktitle="Interactive Storytelling",
year="2019",
publisher="Springer International Publishing",
address="Cham",
pages="304--317",
isbn="978-3-030-33894-7"
}