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