diff --git a/README.md b/README.md index 8db2f05..e409733 100644 --- a/README.md +++ b/README.md @@ -25,11 +25,11 @@ import towhee towhee.glob('./towhee.jpg') \ .image_decode.cv2() \ - .image_embedding.data2vec_vision(model_name='facebook/data2vec-vision-base-ft1k') \ + .image_embedding.data2vec(model_name='facebook/data2vec-vision-base-ft1k') \ .show() ``` -result1 +result1 *Write a same pipeline with explicit inputs/outputs name specifications:* @@ -38,11 +38,11 @@ import towhee towhee.glob['path']('./towhee.jpg') \ .image_decode.cv2['path', 'img']() \ - .image_embedding.data2vec_vision['img', 'vec'](model_name='facebook/data2vec-vision-base-ft1k') \ + .image_embedding.data2vec['img', 'vec'](model_name='facebook/data2vec-vision-base-ft1k') \ .select['img', 'vec']() \ .show() ``` -result2 +result2
@@ -53,7 +53,7 @@ towhee.glob['path']('./towhee.jpg') \ Create the operator via the following factory method -***data2vec_vision(model_name='facebook/data2vec-vision-base')*** +***data2vec(model_name='facebook/data2vec-vision-base')*** **Parameters:** diff --git a/__init__.py b/__init__.py index 5fd45d1..0559bc2 100644 --- a/__init__.py +++ b/__init__.py @@ -15,5 +15,5 @@ from .data2vec_vision import Data2VecVision -def data2vec_vision(model_name='facebook/data2vec-vision-base'): +def data2vec(model_name='facebook/data2vec-vision-base'): return Data2VecVision(model_name)