从CSV文件加载多类图像

Question

我有“训练”和“测试”文件夹，并且在每个文件夹中都有许多文件夹，每个文件夹中都有图像。 在.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

那是正确的做法吗？ 如何为每个文件夹加载图像和相应的标签？

Answer 1

您可以使用inbuild flow_from_directory（）方法查看此链接keras文档