使用keras和python加载并显示dog和cat数据集中的图像

Question

如何使用keras和python、从数据集中返回和显示图像

这是我下载并解压缩数据集的代码

_URL = 'https://storage.googleapis.com/mledu-datasets/cats_and_dogs_filtered.zip'
data_dir = tf.keras.utils.get_file('cats_and_dogs_filterted', origin=_URL, extract=True)
data_dir = pathlib.Path(data_dir)
train_dir = os.path.join(data_dir, 'train')
validation_dir = os.path.join(data_dir, 'validation')
train_dir = pathlib.Path(train_dir)
train_dir

结果是这样的路径：

PosixPath('/root/.keras/datasets/cats_and_dogs_filterted/train')

现在，如果我尝试这个代码：

cats = list(train_dir.glob('cats/*'))
cats

只返回空列表[]

Answer 1

我能够在colab中复制这个问题。解决方法是设置cache_subdir，这是一个绝对路径，文件将保存在该位置。

# Set "cache_subdir" to path-to-folder
data_dir = tf.keras.utils.get_file('cats_and_dogs_filterted', cache_subdir='/content/cats_and_dogs_filterted', origin=_URL, extract=True)

要列出特定文件夹的所有文件，你可以在下面这样做

cats = glob.glob('/content/cats_and_dogs_filterted/cats_and_dogs_filtered/train/cats/*')

请参考本要点检查输出。非常感谢。