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.')