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

Question

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

有与此等效的张量流或 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