Text2image 文本到圖像的合成

文本到圖像合成 用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"

例子,參見如下。



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章