python - 如何在 Tensorflow 中使用现有数据集的一部分

Question

我想在 tensorflow resnet 示例中仅使用 CIFAR10 数据集每类的前 10% code 。

这种情况下，最简单的改变方法是什么？我不太了解Tensorflow中的数据类...

Answer 1

过滤数据集精确每个类 10% 可能有点麻烦，但如果您的目标只是尝试在较小的数据集上使用 ResNet，最简单的方法是skip数据集的 90%，这会将每个类大约减少到 10%。如果这适合您，只需添加 skip after the shuffle call :

...
dataset = dataset.shuffle(buffer_size=_SHUFFLE_BUFFER)
dataset = dataset.skip(54000)
...

..这将导致数据集跳过 60000 个示例中的 90%，仅返回 6000 个示例。