Add audio_embedding_clmr.py

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
2 years ago · de2b0cf173
1 changed files with 11 additions and 0 deletions
--- a/audio_embedding_clmr.py
+++ b/audio_embedding_clmr.py
@ -0,0 +1,11 @@
+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.clmr())
+            .output('vec')
+    )
+