📅  最后修改于: 2022-03-11 14:45:32.535000             🧑  作者: Mango
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