logo
Browse Source

Update

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
shiyu22 3 years ago
parent
commit
8abea24e69
  1. 6
      README.md

6
README.md

@ -1,10 +1,10 @@
# Pipeline: Image Embedding using Resnet
# Pipeline: Image Embedding using Resnet50
Authors: derekdqc, shiyu22 Authors: derekdqc, shiyu22
## Overview ## Overview
The pipeline is used to **extract the feature vector of a given image**. It first normalizes the image and then uses Resnet model to generate the vector.
The pipeline is used to **extract the feature vector of a given image**. It first normalizes the image and then uses Resnet50 model to generate the vector.
## Interface ## Interface
@ -39,7 +39,7 @@ $ pip3 install towhee
>>> from PIL import Image >>> from PIL import Image
>>> img = Image.open('path/to/your/image') # for example, './test.jpg' >>> img = Image.open('path/to/your/image') # for example, './test.jpg'
>>> embedding_pipeline = pipeline('towhee/image-embedding-resnet')
>>> embedding_pipeline = pipeline('towhee/image-embedding-resnet50')
>>> embedding = embedding_pipeline(img) >>> embedding = embedding_pipeline(img)
``` ```

Loading…
Cancel
Save