查看指定目录下的文件名称和序号

import os
directory = "../hospital_data/train"
classes = sorted(entry.name for entry in os.scandir(directory) if entry.is_dir())
if not classes:
    raise FileNotFoundError(f"Couldn't find any class folder in {directory}.")

class_to_idx = {cls_name: i for i, cls_name in enumerate(classes)}
print(classes,class_to_idx)