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
By Denise Braun at Jun 07 2020
By Denise Braun at Jun 07 2020