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
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)
