Browse Source
Add audio embedding
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
audio_embedding.py
|
|
@ -31,6 +31,6 @@ def AudioEmbedding(config=None): |
|
|
return ( |
|
|
return ( |
|
|
pipe.input('path') |
|
|
pipe.input('path') |
|
|
.map('path', 'frame', ops.audio_decode.ffmpeg(config.batch_size, config.sample_rate, config.layout)) |
|
|
.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), conf=op_config) |
|
|
|
|
|
|
|
|
.map('frame', 'vec', ops.audio_embedding.vggish(config.weights_path, config.framework), config=op_config) |
|
|
.output('vec') |
|
|
.output('vec') |
|
|
) |
|
|
) |
|
|
|
