在根据标签类别将images文件夹内的数据集分别存放在以标签命名的子文件夹中,现要读取其标签的方法(或者读取子文件夹名称):
import os
# 数据集存储的根目录(假设数据集存储在images文件夹的子文件夹中)
dataset_root = "./images"
# 读取子文件夹名称作为分类标签
labels = os.listdir(dataset_root)
print("分类标签:", labels)
在根据标签类别将images文件夹内的数据集分别存放在以标签命名的子文件夹中,现要读取其标签的方法(或者读取子文件夹名称):
import os
# 数据集存储的根目录(假设数据集存储在images文件夹的子文件夹中)
dataset_root = "./images"
# 读取子文件夹名称作为分类标签
labels = os.listdir(dataset_root)
print("分类标签:", labels)