我有“训练”和“测试”文件夹,并且每个文件夹中都有许多文件夹,每个文件夹中都包含图像。在.csv文件中,每个文件夹和类都有标签。
这是csv文件
https://i.imgur.com/qMLGOpC.png
和文件夹
https://i.imgur.com/RrYBZxG.png
如何在keras中使用标签加载这些文件夹?
我试图用下面的文件夹和标签制作一个数据框:
pth = 'C:/Users/Documents/train/'
folders = os.listdir(pth)
filepath='C:/Users//Documents//keras/labels.csv'
metadf = pd.read_csv(filepath)
metadf.index = metadf.Class
videos = pd.DataFrame([])
for folder in folders:
pth_upd = pth + folder + '/'
for file in allfiles:
videos = pd.DataFrame(metadf.values, index=folders)
输出为:
https://i.imgur.com/CsXAE8f.png
这是正确的方法吗?如何为每个文件夹加载图像和相应的标签?