def get_dataset_id(data_id):
options_data = dataset.get_yaml_options()
options = dict_to_object(options_data[data_id])
filename = options.filename + "." + options.type
filepath = dataset.learning_data_dir() + os.sep + filename
if os.path.exists(filepath) and os.path.isfile(filepath):
data = dataset.CostmapDataset(filename)
numtrain = data.train_inputs.shape[0]
numtest = data.test_inputs.shape[0]
numdatasets = numtrain + numtest
assert options.numdatasets == numdatasets
assert options.xsize == data.train_targets.shape[1]
assert options.ysize == data.train_targets.shape[2]
assert options.xsize == data.train_inputs.shape[1]
assert options.ysize == data.train_inputs.shape[2]
assert options.xsize == data.test_targets.shape[1]
assert options.ysize == data.test_targets.shape[2]
assert options.xsize == data.test_inputs.shape[1]
assert options.ysize == data.test_inputs.shape[2]
return data
else:
datasets, workspaces = random_environments(options)
dataset.write_dictionary_to_file(datasets, filename)
dataset.write_dictionary_to_file(
workspaces, options.workspaces + "." + options.type)
return get_dataset_id(data_id)
def get_options(self):
""" Load dataset options from file or option parser """
if self._use_parser:
parser = self.environment_parser()
(options, args) = parser.parse_args()
return options
else:
options = dataset.get_yaml_options()[self._dataset_id]
return dict_to_object(options)
dest='seed',
help='Random number seed. -ve values\
mean random seed')
return parser
def get_options(self):
""" Load dataset options from file or option parser """
if self._use_parser:
parser = self.environment_parser()
(options, args) = parser.parse_args()
return options
else:
options = dataset.get_yaml_options()[self._dataset_id]
return dict_to_object(options)
def remove_file_if_exists(file):
if os.path.exists(file):
os.remove(file)
if __name__ == '__main__':
parser = RandomEnvironmentOptions()
options = parser.get_options()
dataset_paramerters = dict_to_object(
dataset.get_yaml_options()[options.dataset_id])
remove_file_if_exists("data/" + dataset_paramerters.filename + ".hdf5")
remove_file_if_exists("data/" + dataset_paramerters.workspaces + ".hdf5")
get_dataset_id(options.dataset_id)