使用预训练的检查点为图像标题-im2txt以零概率获取一个单词作为标题

Question

我正在为im2txt使用预先训练的2M检查点。我得到一个字作为标题如下。我正在使用tensorflow 1.0版Ubuntu 16.04

0)     (p=0.000000)
1)the  (p=0.000000)
2)two  (p=0.000000)

我期待完整的句子，但得到一个不相关的单字标题

Answer 1

在caption_generator.py下

将以下行更改为

        most_likely_words = np.argsort(word_probabilities)[-self.beam_size:][::-1]

如果问题仍然存在，请尝试使用经过最新代码培训的检查点文件。