Python Sampler.beam_search示例

示例#1

文件： routes.py 项目： Leekwanmeng/Image-Captioning

def build_caption_from_sampler(filename):
    saved_model = get_saved_model("checkpoint_2_7369.pt")
    model = Sampler(saved_model, vocab)
    seq, alpha = model.beam_search(filename)
    caption = ""
    for i in seq[1:-1]:
        caption += " " + vocab_idx2word[str(i)]
    return caption, seq, alpha

示例#2

文件： test_sampler.py 项目： Leekwanmeng/Image-Captioning

from sampler import Sampler
from utils import Vocabulary


model = Sampler('./models/checkpoint_2_7369.pt', './data/vocab.pkl')

# model.decoder.max_seg_length = 20


s, seq, alpha = model.beam_search('./sample/sample_beach.jpg')
print(s)
s, seq, alpha = model.beam_search('./sample/sample_img.jpg')
print(s)
s, seq, alpha = model.beam_search('./sample/sample_office.jpg')
print(s)
s, seq, alpha = model.beam_search('./sample/sample_dog.jpg')
print(s)
# print(seq)