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

Question

我正在为 im2txt 使用预先训练的 2M 检查点。我得到一个词作为标题，如下所示。我正在使用张量流版本 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]

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