如何从tensorflow中的子目录导入图像数据集

How to import image dataset from subdirectory in tensorflow

我有一个手语数据集。数据集已存储在如下子目录中。这是因为为相同的标签生成了不同的数据。

此文件夹在 .csv 文件中用 'Sentence' 和 'Path of video' 列注释,其中图像是视频数据集的帧。

每个视频数据集的帧都存储为.jpg 文件。目录结构如下:

|
|---- ALL THE ABOVE --|--- 1 --- filename1.jpg
|                     |      --- filename2.jpg 
|                     |----2 --- filename1.jpg
|                            --- filename2.jpg 
|---- ALL IS WELL -----|----1 --- filename1.jpg
|                      |      --- filename2.jpg
|                      |----2 --- filename1.jpg
|                             --- filename2.jpg

文件夹总数为101,标签总数为22。

但问题是,如何使用 tensorflow 将图像递归导入模型并将此数据文件夹拆分为训练数据集,并在 tensorflow 中测试数据集,我需要从目录中导入每一帧。第二个问题(对不起,如果它很愚蠢,但我需要知道),如何将这个 .csv 文件(由它的标签和文件路径组成)用于模型

如果有人解决我的疑惑和困惑,我将不胜感激。如有语法错误请见谅

谢谢大家

我建议使用以下功能tf.keras.utils.image_dataset_from_directory。您可以选择是否要推断标签。