def get_data_config(self, conf_module):
# get default data config
path = os.path.join(
get_dataset_path(conf_module.configuration['dataset_name']),
'dataset_spec.py')
data_conf_file = imp.load_source('dataset_spec', path)
data_conf = AttrDict()
data_conf.dataset_spec = AttrDict(data_conf_file.dataset_spec)
# update with custom params if available
update_data_conf = {}
if hasattr(conf_module, 'data_config'):
update_data_conf = conf_module.data_config
elif conf_module.configuration.dataset_name is not None:
update_data_conf = importlib.import_module(
'gcp.datasets.configs.' +
conf_module.configuration.dataset_name).config
for key in update_data_conf:
if key == "dataset_spec":
data_conf.dataset_spec.update(update_data_conf.dataset_spec)
else:
data_conf[key] = update_data_conf[key]
if not 'fps' in data_conf:
data_conf.fps = 4
return data_conf
def get_data_config(self, conf_module):
# get default data config
path = os.path.join(
get_dataset_path(conf_module.configuration['dataset_name']),
'dataset_spec.py')
data_conf_file = imp.load_source('dataset_spec', path)
data_conf = AttrDict()
data_conf.dataset_spec = AttrDict(data_conf_file.dataset_spec)
# update with custom params if available
try:
update_data_conf = conf_module.data_config
except AttributeError:
pass
for key in update_data_conf:
if key == "dataset_spec":
data_conf.dataset_spec.update(update_data_conf.dataset_spec)
else:
data_conf[key] = update_data_conf[key]
if not 'fps' in data_conf:
data_conf.fps = 4
return data_conf
