获取文件夹标签

这个 Python 函数用于获取指定文件夹内子文件夹的标签，主要是为了在处理数据时区分不同类型的数据。该函数接收一个参数 Dir，代表指定的文件夹路径。

函数会遍历 Dir 中的子文件夹，排除以点号开头的文件夹，然后根据子文件夹的名字分别返回相应的标签。

以下是该函数的代码片段：

def get_label(Dir):
    for nextdir in os.listdir(Dir):
        if not nextdir.startswith('.'):
            if nextdir in ['NORMAL']:
                label = 0
            elif nextdir in ['PNEUMONIA']:
                label = 1
            else:
                label = 2
            return nextdir, label

其中 nextdir 代表当前遍历到的子文件夹名，label 代表该子文件夹的标签。

返回的值包含两个部分，第一个是子文件夹的名字，第二个是标签。

该函数返回的数据可以用于后续处理，例如用于图像分类任务中的标签设置。