Skip to content

AssertionError when running noc_captioner.py -i images_list.txt #9

Description

@dreamflasher
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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions