从CSV文件加载多类图像

时间:2018-12-19 18:40:06

标签: python pandas keras deep-learning

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

这是正确的方法吗?如何为每个文件夹加载图像和相应的标签?

1 个答案:

答案 0 :(得分:0)

you can use inbuild flow_from_directory() method

查看此链接keras docs