如何获取每个批次图片的读取路径
收藏
自定义数据集加载的时候我应该如何获取每个批次的图片路径，是应该在__getitem__中return，还是有官方的API进行获取路径呢？有没有大佬可以帮我解决一下，救救孩子吧。
0
收藏
请登录后评论
读取数据列表在Dataset的__init__()中手动进行，有没有读取的api得查查文档。
但自己在__getitem__()中返回应该不行，因为return的数据回自动加上paddle.to_tensor()处理，会报错
这是Dataset的api 文档：
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/fluid/dataloader/dataset/Dataset_cn.html#dataset
可以参考下模型库怎么写的，PaddleDetection、PaddleX、PaddleClas这些，翻源码的时候理解最快