文本到圖像合成 用Thought Vectors
這是一個爲從描述合成圖像的tensorflow實現,用的是Skip Thought Vectors模型。
基於DCGAN(tf版本)來構建的。所使用的圖像是GAN-CLS算法所使用的圖像數據。
依賴;
-Python 2.7.6
-Tensorflow
-h5py
-Theano : for skip thought vectors
-scikit-learn : for skip thought vectors
-NLTK : for skip thought vectors
數據集:通過運行python download_datasets.py來下載數據集。
目前的模型是基於flowers dataset訓練的。
預訓練模型文件在Data/skipthoughts下。
訓練:基本用法是python train.py --data_set="flowers"
生成圖像:
python generate_thought_vectors.py --caption_file="Data/sample_captions.txt"
例子,參見如下。