logo
Browse Source

update the readme.

Signed-off-by: wxywb <xy.wang@zilliz.com>
main
wxywb 2 years ago
parent
commit
4255c724a1
  1. 40
      README.md

40
README.md

@ -22,38 +22,28 @@ Load an image from path './teddy.jpg' to generate an image embedding.
Read the text 'Плюшевый мишка на скейтборде на Таймс-сквер.' to generate an text embedding. Read the text 'Плюшевый мишка на скейтборде на Таймс-сквер.' to generate an text embedding.
*Write the pipeline in simplified style*:
*Write a pipeline with explicit inputs/outputs name specifications:*
```python ```python
import towhee import towhee
towhee.glob('./teddy.jpg') \
.image_decode() \
.image_text_embedding.ru_clip(model_name='ruclip-vit-base-patch32-224', modality='image') \
.show()
from towhee.dc2 import pipe, ops, DataCollection
towhee.dc(["'Плюшевый мишка на скейтборде на Таймс-сквер."]) \
.image_text_embedding.ru_clip(model_name='ruclip-vit-base-patch32-224', modality='text') \
.show()
```
<img src="./vec1.png" alt="result1" style="height:20px;"/>
<img src="./vec2.png" alt="result2" style="height:20px;"/>
*Write a same pipeline with explicit inputs/outputs name specifications:*
```python
import towhee
img_pipe = (
pipe.input('url')
.map('url', 'img', ops.image_decode.cv2_rgb())
.map('img', 'vec', ops.image_text_embedding.ru_clip(model_name='ruclip-vit-base-patch32-224', modality='image'))
.output('img', 'vec')
)
towhee.glob['path']('./teddy.jpg') \
.image_decode['path', 'img']() \
.image_text_embedding.ru_clip['img', 'vec'](model_name='ruclip-vit-base-patch32-224', modality='image') \
.select['img', 'vec']() \
.show()
text_pipe = (
pipe.input('text')
.map('text', 'vec', ops.image_text_embedding.ru_clip(model_name='ruclip-vit-base-patch32-224', modality='text'))
.output('text', 'vec')
)
towhee.dc['text'](["Плюшевый мишка на скейтборде на Таймс-сквер."]) \
.image_text_embedding.ru_clip['text','vec'](model_name='ruclip-vit-base-patch32-224', modality='text') \
.select['text', 'vec']() \
.show()
DataCollection(img_pipe('./teddy.jpg')).show()
DataCollection(text_pipe('Плюшевый мишка на скейтборде на Таймс-сквер.')).show()
``` ```
<img src="./tabular1.png" alt="result1" style="height:60px;"/> <img src="./tabular1.png" alt="result1" style="height:60px;"/>
<img src="./tabular2.png" alt="result2" style="height:60px;"/> <img src="./tabular2.png" alt="result2" style="height:60px;"/>

Loading…
Cancel
Save