diff --git a/README.md b/README.md index 4569711..544e653 100644 --- a/README.md +++ b/README.md @@ -1,10 +1,10 @@ -# Pipeline: Image Embedding using Resnet +# Pipeline: Image Embedding using Resnet50 Authors: derekdqc, shiyu22 ## 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 @@ -39,7 +39,7 @@ $ pip3 install towhee >>> from PIL import Image >>> 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) ```