logo
Browse Source

change data2vecvision to data2vec.

Signed-off-by: wxywb <xy.wang@zilliz.com>
main
wxywb 2 years ago
parent
commit
72e04b699f
  1. 10
      README.md
  2. 2
      __init__.py

10
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()
```
<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:**

2
__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)

Loading…
Cancel
Save