Add yaml

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
3 years ago · fd74966b27
1 changed files with 70 additions and 0 deletions
--- a/audio_embedding_vggish.yaml
+++ b/audio_embedding_vggish.yaml
@ -0,0 +1,70 @@
 name: 'audio_embedding_vggish'
 operators:
    -
        name: '_start_op'
        function: '_start_op'
        init_args:
        inputs:
            -
                df: '_start_df'
                name: 'filepath'
                col: 0
        outputs:
            -
                df: 'audio'
        iter_info:
            type: map
    -
        name: 'embedding_model'
        function: 'towhee/tf_vggish_audioset'
        init_args:
            sample_rate: 16000
        inputs:
            -
                df: 'audio'
                name: 'filepath'
                col: 0
        outputs:
            -
                df: 'emb'
        iter_info:
            type: map
    -
        name: '_end_op'
        function: '_end_op'
        init_args:
        inputs:
            -
                df: 'emb'
                name: 'embs'
                col: 0
        outputs:
            -
                df: '_end_df'
        iter_info:
            type: map
 dataframes:
    -
        name: '_start_df'
        columns:
            -
                name: 'filepath'
                vtype: 'str'
    -
        name: 'audio'
        columns:
            -
                name: 'filepath'
                vtype: 'str'
    -
        name: 'emb'
        columns:
            -
                name: 'embs'
                vtype: 'numpy.ndarray'
    -
        name: '_end_df'
        columns:
            -
                name: 'embs'
                vtype: 'numpy.ndarray'