tensorflow如何加载‘kfold划分的’图像数据集

tensorflow如何加载‘kfold划分的’图像数据集 ？

已知：有图像数据和另一组数据（完全对应），根据另一组数据进行kfold划分，得到train_index和test_index.

   已有：（1）一个包含所有图像数据的文件夹 ‘。/Image/’；
        （2）一个包含file name + 标签label的dataY csv文件‘dataY.csv'
   目的：使用 【tensoflow】 构建一个类，可以根据kfold得到的index + dataY.csv 取Image图像，从而进行图像分类

目前尝试过的：

使用 pytorch构建自己的torch.utils.data.Dataset类，修改‘__init__’ ‘__getitem__‘即可实现

处于一些限制，我希望用tensorflow达到这样的目的，但是找不到对应可用的方法，大部分教程都是一开始划分好 类的根目录，划分好 【train】【validation】的根目录，万分感谢！

不用那么麻烦，直接用sklearn的kfold就行，kfold无非就是个随机分数据集训练而已。。

from sklearn.model_selection import KFold
kfold = KFold(n_splits=num_folds, shuffle=True)

for train, test in kfold.split(inputs, targets):
    #下面跟tensorflow的模型训练代码就行了