时间戳

首页 > 解决方案 > 从 csv 文件加载多类图像

python - 从 csv 文件加载多类图像

问题描述

我有训练和测试文件夹，每个文件夹里面都有很多文件夹，每个文件夹里面都有图像。在 .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

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

标签： pythonpandaskerasdeep-learning

解决方案

您可以使用 inbuild flow_from_directory() 方法查看此链接keras docs

推荐阅读