add audio_embedding_vggish.py

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
3 years ago · f690ed3f78
1 changed files with 10 additions and 0 deletions
--- a/audio_embedding_vggish.py
+++ b/audio_embedding_vggish.py
@ -0,0 +1,10 @@
+from towhee import pipe, ops, AutoPipes
+
+@AutoPipes.register
+def AudioEmbeddingVggish():
+    return (
+        pipe.input('path')
+            .map('path', 'frame', ops.audio_decode.ffmpeg())
+            .map('frame', 'vec', ops.audio_embedding.vggish())
+            .output('vec')
+    )