Computing features for images 0-9 of 10
Traceback (most recent call last):
File "noc_captioner.py", line 569, in <module>
main()
File "noc_captioner.py", line 553, in main
strategies=STRATEGIES, display_vocab=vocab_list)
File "noc_captioner.py", line 286, in run_pred_iters
run_pred_iter(pred_net, image_fc7, display_vocab, strategies=strategies)
File "noc_captioner.py", line 199, in run_pred_iter
captions, probs = predict_image_caption(net, mean_pool_fc7, vocab_list, strategy=strategy)
File "noc_captioner.py", line 97, in predict_image_caption
return predict_image_caption_beam_search(net, mean_pool_fc7, vocab_list, strategy)
File "noc_captioner.py", line 174, in predict_image_caption_beam_search
assert probs.shape[0] == len(vocab_list)
AssertionError