name: 'image-embedding-resnet' operators: - name: '_start_op' function: '_start_op' init_args: inputs: - df: '_start_df' name: 'img_path' col: 0 outputs: - df: 'image' iter_info: type: map - name: 'embedding_model' function: 'towhee/resnet-image-embedding' # same as 'resnet50-image-embedding', default user is towhee tag: 'main' # optional init_args: model_name: 'resnet50' inputs: - df: 'image' name: 'img_path' col: 0 outputs: - df: 'embedding' iter_info: type: map - name: '_end_op' function: '_end_op' init_args: inputs: - df: 'embedding' name: 'feature_vector' col: 0 outputs: - df: '_end_df' iter_info: type: map dataframes: - name: '_start_df' columns: - name: 'img_path' vtype: 'str' - name: 'image' columns: - name: 'img_path' vtype: 'str' - name: 'embedding' columns: - name: 'feature_vector' vtype: 'numpy.ndarray' - name: '_end_df' columns: - name: 'feature_vector' vtype: 'numpy.ndarray'