从我的硬盘加载数据集的python代码是什么?

Question

我尝试从TensorFlow官方教程学习机器学习。

但是大多数教程都是在命令提示符下下载的。

我找不到任何关于从我自己的磁盘加载我自己的图像数据集的教程。

如果你能给我一个直接的答复就太好了。

我把图像数据集放在我的window 10桌面:

C:UsersUserDesktopDataSetcoinsdata

test (label 1-211)
train (label 1-211)
validation (label 1-211)

Answer 1

你可以使用image_dataset_from_directory，你只需要在参数directory中传递路径到文件。

from tensorflow.keras.preprocessing import image_dataset_from_directory
train_dataset = image_dataset_from_directory(
directory=TRAIN_DIR,
labels="inferred",
label_mode="categorical",
image_size=SIZE,
seed=SEED,
subset=None,
interpolation="bilinear",
follow_links=False,
)
validation_dataset = image_dataset_from_directory(
directory=VALIDATION_DIR,
labels="inferred",
label_mode="categorical",
image_size=SIZE,
seed=SEED,
subset=None,
interpolation="bilinear",
follow_links=False,
)
test_dataset = image_dataset_from_directory(
directory=TEST_DIR,
labels="inferred",
label_mode="categorical",
image_size=SIZE,
seed=SEED,
subset=None,
interpolation="bilinear",
follow_links=False,
)

Answer 2

您可以在keras中使用flow_from_disk。这是一个很好的教程

keras中磁盘的流