Dataset用load_dataset读图片和对应的caption的一个坑

代码:

data_files = {}
if args.train_data_dir is not None:
     data_files["train"] = os.path.join(args.train_data_dir, "**")
     dataset = load_dataset(
            "imagefolder",
            data_files=data_files,
            cache_dir=args.cache_dir,
        )

数据:

想要读进图片和以及对应的caption, 结果只是读出来了和label(这里只有一个文件夹,所以label全是0)。   于是一直找错,查看huggingface官网:https://huggingface.co/docs/datasets/v2.4.0/en/image_load#imagefolder

自己代码和他写的一摸一样,为什么会把caption读不进去呢?    

终于。。。。。。

发现了我的 datasets的版本是2.1.0。     发现我浏览的huaggingface官网的版本是2.4.0。   好的,立马换了版本,over,成功读如图片和caption!!!!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值