负载张量流数据集“斯坦福狗”

2024-04-19 08:14:31 发布

您现在位置:Python中文网/ 问答频道 /正文

我想使用tensor\u flow\u数据集集合的“stanford\u dogs”数据集。 如下所示: https://www.tensorflow.org/datasets/catalog/stanford_dogs

但是当尝试加载时,它不会列在可用的数据集中。 如何加载“斯坦福狗”

import tensorflow_datasets as tfds

SPLIT_WEIGHTS = (8, 1, 1)
splits = tfds.Split.TRAIN.subsplit(weighted=SPLIT_WEIGHTS)
(raw_train, raw_validation, raw_test), metadata = tfds.load(name="stanford_dogs", 
                                                            with_info=True,
                                                            split=list(splits),
                                                             batch_size=-1,
                                                            as_supervised=True)

Tags: 数据httpstruerawtensorflowasflowdatasets