Browse Source
change data2vecvision to data2vec.
Signed-off-by: wxywb <xy.wang@zilliz.com>
main
wxywb
3 years ago
2 changed files with
6 additions and
6 deletions
-
README.md
-
__init__.py
|
|
@ -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() |
|
|
|
|
|
|
|
``` |
|
|
|
<img src="https://towhee.io/image-embedding/data2vec-vision/raw/branch/main/result1.png" alt="result1" style="height:20px;"/> |
|
|
|
<img src="./result1.png" alt="result1" style="height:20px;"/> |
|
|
|
|
|
|
|
*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() |
|
|
|
``` |
|
|
|
<img src="https://towhee.io/image-embedding/data2vec-vision/raw/branch/main/result2.png" alt="result2" style="height:60px;"/> |
|
|
|
<img src="./result2.png" alt="result2" style="height:60px;"/> |
|
|
|
|
|
|
|
|
|
|
|
<br /> |
|
|
@ -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:** |
|
|
|
|
|
|
|
|
|
@ -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) |
|
|
|