python - 如何从 tensorflow/keras 中下载的 tar.gz 文件加载数据？

Question

Tensorflow 数据集或 tfds 会自动开始下载我想要的数据。我在我的系统中下载了 cifar10。我可以使用以下命令直接在 pytorch 中加载数据:
torchvision.datasets.CIFAR10('path/to/directory',...,download=False)

是否有与此等效的 tensorflow 或 keras？

Answer 1

我认为你能做的最好的事情是首先提取 tar 文件:

import tarfile

if fname.endswith("tar.gz"):
    tar = tarfile.open(fname, "r:gz")
    tar.extractall()
    tar.close()
elif fname.endswith("tar"):
    tar = tarfile.open(fname, "r:")
    tar.extractall()
    tar.close()

然后访问模型数据并使用 keras 加载它:
https://www.tensorflow.org/api_docs/python/tf/keras/models/load_model