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

0

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

By Beginner Denise Braun at Jun 07 2020


Related code examples


Code examples by languages