keras 如何输入目录以列出所有子目录?

au9on6nz  于 2023-03-23  发布在  其他
关注(0)|答案(2)|浏览(120)

我被一个错误卡住了。“Keras_merged_train”目录有两个名为Deepfake和真实的的子目录。我想使用glob函数来计算“Keras_merged_train”目录中文件夹的长度(答案应该是2)。我在打印后发现glob并没有进入目录,只是打印了目录的名称。如何解决这个问题?

这是我试过的。长度应该是2,但它没有阅读里面的文件。

里面的文件。

ej83mcc0

ej83mcc01#

glob默认不进入目录,必须追加/*

keras_merged_train_folders = glob('/content/Keras_merged_train/*')

也可以使用pathlib

import pathlib

data_dir = pathlib.Path('/content/Keras_merged_train')

keras_merged_train_folders = [str(p) for p in data_dir.iterdir() if p.is_dir()]
j9per5c4

j9per5c42#

您可以使用os获取目录中存在的文件和文件夹的名称,然后对其进行计数。下面是一个示例:

import os
files = os.listdir('/path/to/directory/')
num_files = len(files)

您可以阅读the documentation以了解有关此库的更多信息。

相关问题