使用tensorflow、keras在python中加载GTZAN数据集

Question

嗨，我在python中使用keras，我是机器学习的新手 我想加载数据，我有一个文件夹，在这个文件夹中我有子文件夹 在子文件夹中，我有一系列音频文件 例如：

（我有10个子文件夹，每个子文件夹都是流派，在子文件夹中我有100个音频文件Spectomga（jpg），连接到流派）

数据->

   pop-> 1.jpg
         2.jpg
         ....
   
  hip hop-> 1.jpg
         2.jpg
         ....
   
  blues-> 1.jpg
         2.jpg
         ....

如何加载数据并拆分

（试一试，（x_列，y_列），（x_测试，y_测试）=？？）

Answer 1

从tensorflow指南tensorflow.org/tutorials/load_data/images：

train_ds = tf.keras.preprocessing.image_dataset_from_directory(
  data_dir,
  validation_split=0.2,
  subset="training",
  seed=123,
  image_size=(img_height, img_width),
  batch_size=batch_size)

获得验证

val_ds = tf.keras.preprocessing.image_dataset_from_directory(
  data_dir,
  validation_split=0.2,
  subset="validation",
  seed=123,
  image_size=(img_height, img_width),
  batch_size=batch_size)

得到标签

train_ds.class_names