0
点赞
收藏
分享

微信扫一扫

在flickr30k上训练时出现KeyError:‘filepath‘报错(imagecaptioning.pytorch-master)

爪哇驿站 2022-04-30 阅读 14

训练命令:

python scripts/prepro_feats.py --input_json data/dataset_flickr30k.json --output_dir data/flickr30talk --images_root flickr30k-images

报错:

在这里插入图片描述
问题原因:
是因为程序没有找到你的flickr30k的图片位置

解决方法:
1、修改dataset_flickr30k.json文件添加‘filepath’这一内容,其对应的是你flickr30k数据集的路径
2、在prepro_feats.py中找到下面的程序

    for i,img in enumerate(imgs):
        # load the image
        I = skimage.io.imread(os.path.join(params['images_root'], img['filepath'], img['filename']))
        # handle grayscale input images

将img[‘filepath’]直接用你的路径替换掉‘路径’

举报

相关推荐

0 条评论