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

季南枫(JIM)

于 2022-04-30 14:55:37 发布

阅读量771

点赞数 1

CC 4.0 BY-SA版权

文章标签： pytorch 深度学习 python

本文链接：https://blog.youkuaiyun.com/weixin_46235937/article/details/124512927

本文指导如何修复python脚本在处理Flickr30k数据集时的图片路径错误，需修改json文件并调整代码以指向正确图片路径，确保图像读取顺利。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

训练命令：

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’]直接用你的路径替换掉‘路径’