From 8abea24e690c632b54be9a89c9db29cea89a00a4 Mon Sep 17 00:00:00 2001 From: shiyu22 Date: Wed, 15 Dec 2021 21:30:30 +0800 Subject: [PATCH] Update Signed-off-by: shiyu22 --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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) ```