implementation in tensorflow 1.4. (also tested on tf 1.0). It is suggested to use tf 1.4 for this code.
- Pre-trained models coming soon.
- Further clean the code soon.
if this code helps you, thanks for citing our work.
@inproceedings{wang2017diverse, title={Diverse and Accurate Image Description Using a Variational Auto-Encoder with an Additive Gaussian Encoding Space}, author={Wang, Liwei and Schwing, Alexander and Lazebnik, Svetlana}, booktitle={Advances in Neural Information Processing Systems}, pages={5758--5768}, year={2017} }