How can I get 'word_table.pickle'? about image_captioning HOT 2 OPEN

The 'word_table.pickle' file is generated automatically when you run the code. The problem is that the directory './words' does not exist. Actually, several directories do not exist and are not checked for. I added the following lines to main.py so that it would automatically create those directories if they don't already exist.

def makeDirs(my_dir):
    if not os.path.exists(my_dir):
        os.makedirs(my_dir)

makeDirs(args.val_result_dir)
makeDirs(args.test_result_dir)
makeDirs(args.glove_dir)
makeDirs(args.save_dir)
makeDirs(os.path.join(args.save_dir, args.cnn_model+'/'))

Don't forget that you'll also need to download the MS COCO datasets.

from image_captioning.

@loganlebanoff
hi, which lines in main.py should the codes above be added?
Thank you

from image_captioning.