brails.processors.year_built_classifier.lib.datasets module

class brails.processors.year_built_classifier.lib.datasets.YearBuiltFolder(image_folder, soft_labels=False, gaussian_std=1.5, transforms=None, classes=None, calc_perf=False)

Bases: Dataset

loader(path)

soft_labels

if not os.path.isdir(image_folder):

if os.path.isfile(image_folder):

# The following format is to be consistent with os.walk output file_list = [[os.path.split(image_folder)[0],’None’,[os.path.split(image_folder)[1]]]]

else:

print(‘Error: Image folder or file {} not found.’.format(image_folder)) exit()

else:

file_list = os.walk(image_folder, followlinks=True)