def _construct_and_fill_model(self):
model_dir = sly.TaskPaths(determine_in_project=False).model_dir
self.device_ids = sly.remap_gpu_devices([self.source_gpu_device])
if 'model.pb' not in os.listdir(model_dir):
logger.info('Freezing training checkpoint!')
freeze_graph('image_tensor', model_dir + '/model.config',
model_dir + '/model_weights/model.ckpt', model_dir)
self.detection_graph = create_detection_graph(model_dir)
self.session = tf.Session(graph=self.detection_graph)
logger.info('Weights are loaded.')
def _construct_and_fill_model(self):
self.device_ids = sly.remap_gpu_devices(self.config['gpu_devices'])
if 'model.pb' not in os.listdir(self.helper.paths.model_dir):
logger.info('Freezing training checkpoint!')
freeze_graph(
'image_tensor', self.helper.paths.model_dir + '/model.config',
self.helper.paths.model_dir + '/model_weights/model.ckpt',
self.helper.paths.model_dir)
self.detection_graph = create_detection_graph(
self.helper.paths.model_dir)
self.session = tf.Session(graph=self.detection_graph)
logger.info('Weights are loaded.')
def _construct_and_fill_model(self):
model_dir = sly.TaskPaths(determine_in_project=False).model_dir
self.device_ids = sly.remap_gpu_devices([self.source_gpu_device])
self.detection_graph = create_detection_graph(model_dir)
self.session = tf.Session(graph=self.detection_graph)
logger.info('Weights are loaded.')
def _construct_and_fill_model(self):
self.device_ids = sly.remap_gpu_devices(self.config['gpu_devices'])
self.detection_graph = create_detection_graph(self.helper.paths.model_dir)
self.session = tf.Session(graph=self.detection_graph)
logger.info('Weights are loaded.')