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)