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



hi im在python中使用keras,在机器学习中使用im new我想加载数据,我有一个文件夹,在这个文件夹中我有子文件夹在子文件夹中,我有音频文件的spectogram示例:

(我有10个子文件夹,每个子文件夹都是流派,在子文件夹里我有100个音频文件spectograma(jpg(,连接到流派(

数据->

pop-> 1.jpg
2.jpg
....

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

blues-> 1.jpg
2.jpg
....

如何加载数据并进行拆分?

(try,(x_train,y_train(,(x_test,y_test(=???(

来自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

相关内容

  • 没有找到相关文章

最新更新