Python 获取数据集名称和分类：使用 os 和 glob 模块

要获取数据集的名称和分类，你可以使用'os'模块和'glob'模块来遍历数据集目录，并解析目录名称来获取分类信息。

以下是一个示例代码，展示了如何获取数据集的名称和分类：

import os
import glob

dataset_dir = 'path/to/your/dataset'

# 获取数据集名称
dataset_name = os.path.basename(dataset_dir)

# 遍历每个目录并获取分类信息
categories = []
for category_dir in glob.glob(os.path.join(dataset_dir, '*')):
    if os.path.isdir(category_dir):
        category_name = os.path.basename(category_dir)
        categories.append(category_name)

print(f"Dataset Name: {dataset_name}")
print(f"Categories: {categories}")

在上述代码中，我们首先使用'os.path.basename()'函数获取数据集目录的名称，并将其保存在'dataset_name'变量中。

然后，我们使用'glob.glob()'函数找到数据集目录中的所有子目录。对于每个子目录，我们使用'os.path.isdir()'函数检查是否为有效的目录。如果是有效的目录，我们使用'os.path.basename()'函数获取目录的名称，并将其保存在'category_name'变量中。然后，将目录名称添加到'categories'列表中。

最后，我们打印出数据集的名称和分类信息。

请将'dataset_dir'替换为你的数据集目录的实际路径。运行上述代码，你将获得数据集的名称和分类信息。