From dd8e14ef073e96c674f4bf20366268efdef8cd02 Mon Sep 17 00:00:00 2001 From: wxywb Date: Mon, 24 Oct 2022 19:52:16 +0800 Subject: [PATCH] amend the requirement. Signed-off-by: wxywb --- README.md | 4 ++-- requirements.txt | 3 +++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index a832f4a..987c5c8 100644 --- a/README.md +++ b/README.md @@ -64,14 +64,14 @@ Create the operator via the following factory method ## 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:** ​ ***img:*** *towhee.types.Image (a sub-class of numpy.ndarray)* -​ The image to generate embedding. +​ The image to generate caption. diff --git a/requirements.txt b/requirements.txt index ca9f2ee..f4f40e9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,3 +3,6 @@ torch torchvision timm towhee +ftfy +yacs +numpy