用keras和python加载和显示来自狗和猫数据集的图像

omhiaaxx  于 2022-11-13  发布在  Python
关注(0)|答案(1)|浏览(144)

如何使用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

仅返回空列表[]

nbewdwxp

nbewdwxp1#

我能够在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/*')

请参考此gist以检查输出。谢谢!

相关问题