logo
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions

Updated 1 year ago

towhee

Pipeline: Image Embedding using resnet50

Authors: Filip

Overview

The pipeline is used to extract the feature vector of a given image. It uses the the resnet50 model from Ross Wightman's timm to generate the vector.

Interface

Input Arguments:

  • img_path:
    • the input image path
    • supported types: str

Pipeline Output:

The pipeline returns a tuple Tuple[('feature_vector', numpy.ndarray)] containing following fields:

  • feature_vector:
    • the embedding of input image
    • data type: numpy.ndarray
    • shape: (1, 2048)

How to use

  1. Install Towhee
$ pip3 install towhee

You can refer to Getting Started with Towhee for more details. If you have any questions, you can submit an issue to the towhee repository.

  1. Run it with Towhee
>>> from towhee import pipeline

>>> img_path = 'path/to/your/image'
>>> embedding_pipeline = pipeline('towhee/image-embedding-resnet50')
>>> embedding = embedding_pipeline(img_path)
shiyu22 8c76aa4bdc remove autoconfig 31 Commits
folder-icon readme_res Update 2 years ago
file-icon .gitignore
3.0 KiB
download-icon
update unittest 2 years ago
file-icon README.md
1.1 KiB
download-icon
upload 2 years ago
file-icon image_embedding_resnet50.py
287 B
download-icon
remove autoconfig 1 year ago
file-icon image_embedding_resnet50.yaml
1.2 KiB
download-icon
upload 2 years ago