def testCreateInferenceGraph(self): with self.test_session() as sess: freeze.create_inference_graph('a,b,c,d', 16000, 1000.0, 30.0, 30.0, 10.0, 40, 'conv') self.assertIsNotNone(sess.graph.get_tensor_by_name('wav_data:0')) self.assertIsNotNone( sess.graph.get_tensor_by_name('decoded_sample_data:0')) self.assertIsNotNone(sess.graph.get_tensor_by_name('labels_softmax:0'))
def testCreateInferenceGraph(self): with self.test_session() as sess: freeze.create_inference_graph('a,b,c,d', 16000, 1000.0, 20.0, 10.0, 40, 'conv') self.assertIsNotNone(sess.graph.get_tensor_by_name('wav_data:0')) self.assertIsNotNone( sess.graph.get_tensor_by_name('decoded_sample_data:0')) self.assertIsNotNone( sess.graph.get_tensor_by_name('labels_softmax:0'))
def testCreateInferenceGraphWithMicro(self): with self.cached_session() as sess: freeze.create_inference_graph(wanted_words='a,b,c,d', sample_rate=16000, clip_duration_ms=1000.0, clip_stride_ms=30.0, window_size_ms=30.0, window_stride_ms=10.0, feature_bin_count=40, model_architecture='conv', preprocess='micro') self.assertIsNotNone(sess.graph.get_tensor_by_name('wav_data:0')) self.assertIsNotNone( sess.graph.get_tensor_by_name('decoded_sample_data:0')) self.assertIsNotNone( sess.graph.get_tensor_by_name('labels_softmax:0'))
def testCreateInferenceGraphWithMicro(self): with self.cached_session() as sess: freeze.create_inference_graph( wanted_words='a,b,c,d', sample_rate=16000, clip_duration_ms=1000.0, clip_stride_ms=30.0, window_size_ms=30.0, window_stride_ms=10.0, feature_bin_count=40, model_architecture='conv', preprocess='micro') self.assertIsNotNone(sess.graph.get_tensor_by_name('wav_data:0')) self.assertIsNotNone( sess.graph.get_tensor_by_name('decoded_sample_data:0')) self.assertIsNotNone(sess.graph.get_tensor_by_name('labels_softmax:0'))
def testFeatureBinCount(self): with self.cached_session() as sess: freeze.create_inference_graph(wanted_words='a,b,c,d', sample_rate=16000, clip_duration_ms=1000.0, clip_stride_ms=30.0, window_size_ms=30.0, window_stride_ms=10.0, feature_bin_count=80, model_architecture='conv', preprocess='average') self.assertIsNotNone(sess.graph.get_tensor_by_name('wav_data:0')) self.assertIsNotNone( sess.graph.get_tensor_by_name('decoded_sample_data:0')) self.assertIsNotNone( sess.graph.get_tensor_by_name('labels_softmax:0')) ops = [node.op for node in sess.graph_def.node] self.assertEqual(0, ops.count('Mfcc'))
def testFeatureBinCount(self): with self.cached_session() as sess: freeze.create_inference_graph( wanted_words='a,b,c,d', sample_rate=16000, clip_duration_ms=1000.0, clip_stride_ms=30.0, window_size_ms=30.0, window_stride_ms=10.0, feature_bin_count=80, model_architecture='conv', preprocess='average') self.assertIsNotNone(sess.graph.get_tensor_by_name('wav_data:0')) self.assertIsNotNone( sess.graph.get_tensor_by_name('decoded_sample_data:0')) self.assertIsNotNone(sess.graph.get_tensor_by_name('labels_softmax:0')) ops = [node.op for node in sess.graph_def.node] self.assertEqual(0, ops.count('Mfcc'))
def testCreateSavedModel(self): tmp_dir = self.get_temp_dir() saved_model_path = os.path.join(tmp_dir, 'saved_model') with self.cached_session() as sess: input_tensor, output_tensor = freeze.create_inference_graph( wanted_words='a,b,c,d', sample_rate=16000, clip_duration_ms=1000.0, clip_stride_ms=30.0, window_size_ms=30.0, window_stride_ms=10.0, feature_bin_count=40, model_architecture='conv', preprocess='micro') global_variables_initializer().run() graph_util.convert_variables_to_constants(sess, sess.graph_def, ['labels_softmax']) freeze.save_saved_model(saved_model_path, sess, input_tensor, output_tensor)