def _load_train_config(self):
train_config_rw = TrainConfigRW(self.helper.paths.model_dir)
if not train_config_rw.train_config_exists:
raise RuntimeError(
'Unable to run inference, config from training wasn\'t found.')
self.train_config = train_config_rw.load()
logger.info(
'Model input size is read (for auto-rescale).',
extra={
'input_size': {
'width': 1200,
'height':
1200 # input shape is fixed for Faster with NasNet encoder
}
})
self.class_title_to_idx = self.train_config['mapping']
self.train_classes = sly.FigClasses(self.train_config['classes'])
logger.info('Read model internal class mapping',
extra={'class_mapping': self.class_title_to_idx})
logger.info('Read model out classes',
extra={'classes': self.train_classes.py_container})
out_class_mapping = {
x: self.class_title_to_idx[x]
for x in (x['title'] for x in self.train_classes)
}
self.inv_mapping = inverse_mapping(out_class_mapping)
def _load_train_config(self):
train_config_rw = TrainConfigRW(self.helper.paths.model_dir)
if not train_config_rw.train_config_exists:
raise RuntimeError(
'Unable to run inference, config from training wasn\'t found.')
self.train_config = train_config_rw.load()
self.class_title_to_idx = self.train_config['mapping']
self.train_classes = sly.FigClasses(self.train_config['classes'])
logger.info('Read model internal class mapping',
extra={'class_mapping': self.class_title_to_idx})
logger.info('Read model out classes',
extra={'classes': self.train_classes.py_container})
out_class_mapping = {
x: self.class_title_to_idx[x]
for x in (x['title'] for x in self.train_classes)
}
self.inv_mapping = inverse_mapping(out_class_mapping)
def _load_train_config(self):
model_dir = sly.TaskPaths(determine_in_project=False).model_dir
train_config_rw = TrainConfigRW(model_dir)
if not train_config_rw.train_config_exists:
raise RuntimeError('Unable to run inference, config from training wasn\'t found.')
self.train_config = train_config_rw.load()
input_size = self.train_config['settings']['input_size']
w, h = input_size['width'], input_size['height']
logger.info('Model input size is read (for auto-rescale).', extra={'input_size': {
'width': w, 'height': h
}})
self.class_title_to_idx = self.train_config['mapping']
self.train_classes = sly.FigClasses(self.train_config['classes'])
logger.info('Read model internal class mapping', extra={'class_mapping': self.class_title_to_idx})
logger.info('Read model out classes', extra={'classes': self.train_classes.py_container})
out_class_mapping = {x: self.class_title_to_idx[x] for x in
(x['title'] for x in self.train_classes)}
self.inv_mapping = inverse_mapping(out_class_mapping)