Browse Source
amend the requirement.
Signed-off-by: wxywb <xy.wang@zilliz.com>
main
wxywb
2 years ago
2 changed files with
5 additions and
2 deletions
-
README.md
-
requirements.txt
|
|
@ -64,14 +64,14 @@ Create the operator via the following factory method |
|
|
|
|
|
|
|
|
## Interface |
|
|
## Interface |
|
|
|
|
|
|
|
|
An image-text embedding operator takes a [towhee image](link/to/towhee/image/api/doc) as input and generate the correspoing caption. |
|
|
|
|
|
|
|
|
An image captioning operator takes a [towhee image](link/to/towhee/image/api/doc) as input and generate the correspoing caption. |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
**Parameters:** |
|
|
**Parameters:** |
|
|
|
|
|
|
|
|
***img:*** *towhee.types.Image (a sub-class of numpy.ndarray)* |
|
|
***img:*** *towhee.types.Image (a sub-class of numpy.ndarray)* |
|
|
|
|
|
|
|
|
The image to generate embedding. |
|
|
|
|
|
|
|
|
The image to generate caption. |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -3,3 +3,6 @@ torch |
|
|
torchvision |
|
|
torchvision |
|
|
timm |
|
|
timm |
|
|
towhee |
|
|
towhee |
|
|
|
|
|
ftfy |
|
|
|
|
|
yacs |
|
|
|
|
|
numpy |
|
|
|
