def setUpClass(self): self.tmp_dir = tempfile.mkdtemp() self.dataset_dir = os.path.join(self.tmp_dir, 'dataset') create_mock_dataset(self.dataset_dir, 160) self.lfw_pairs_file = create_mock_lfw_pairs(self.tmp_dir) print(self.lfw_pairs_file) self.pretrained_model_name = '20180402-114759' download_and_extract.download_and_extract_file(self.pretrained_model_name, 'data/') download_and_extract.download_and_extract_file('lfw-subset', 'data/') self.model_file = os.path.join('data', self.pretrained_model_name, 'model-%s.ckpt-275' % self.pretrained_model_name) self.pretrained_model = os.path.join('data', self.pretrained_model_name) self.frozen_graph_filename = os.path.join('data', self.pretrained_model_name+'.pb') print('Memory utilization (SetUpClass): %.3f MB' % memory_usage_psutil())
def setUpClass(self): self.tmp_dir = tempfile.mkdtemp() self.dataset_dir = os.path.join(self.tmp_dir, 'dataset') create_mock_dataset(self.dataset_dir, 160) self.lfw_pairs_file = create_mock_lfw_pairs(self.tmp_dir) print(self.lfw_pairs_file) self.pretrained_model_name = '20180402-114759' download_and_extract.download_and_extract_file( self.pretrained_model_name, 'data/') download_and_extract.download_and_extract_file('lfw-subset', 'data/') self.model_file = os.path.join( 'data', self.pretrained_model_name, 'model-%s.ckpt-275' % self.pretrained_model_name) self.pretrained_model = os.path.join('data', self.pretrained_model_name) self.frozen_graph_filename = os.path.join( 'data', self.pretrained_model_name + '.pb') print('Memory utilization (SetUpClass): %.3f MB' % memory_usage_psutil())
def download_model(model, path): de.download_and_extract_file(model, path)