From 376a8e6c45ed592e6073f1291894c606509b2fe6 Mon Sep 17 00:00:00 2001 From: shiyu22 Date: Mon, 27 Mar 2023 16:22:28 +0800 Subject: [PATCH] Add audio embedding Signed-off-by: shiyu22 --- audio_embedding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/audio_embedding.py b/audio_embedding.py index eb9cb9c..aaf1f33 100644 --- a/audio_embedding.py +++ b/audio_embedding.py @@ -30,7 +30,7 @@ def AudioEmbedding(config=None): return ( pipe.input('path') - .map('path', 'frame', ops.audio_decode.ffmpeg(config.batch_size, config.sample_rate, config.layout)) - .map('frame', 'vec', ops.audio_embedding.vggish(config.weights_path, config.framework), config=op_config) + .map('path', 'frame', ops.audio_decode.ffmpeg(batch_size=config.batch_size, sample_rate=config.sample_rate, layout=config.layout)) + .map('frame', 'vec', ops.audio_embedding.vggish(weights_path=config.weights_path, framework=config.framework), config=op_config) .output('vec') )