def _create_dataset(self, set_name):
dim = (self.dataset_config.input_dim, self.dataset_config.output_dim)
path = self.dataset_path
preprocessing = self.dataset_config.use_preprocessing
print("---- Using preprossing: {}".format(preprocessing))
std = self.dataset_config.dataset_std
samples_per_image = 200
creator = Creator(path, dim=dim, preproccessing=preprocessing, std=std)
creator.load_dataset()
#Creating a shared variable of sampled test data
raw_set = None
if set_name == "valid":
raw_set = creator.valid
else:
raw_set = creator.test
aerial = AerialDataset()
return aerial.shared_dataset(creator.sample_data(
raw_set, samples_per_image),
cast_to_int=True)
