diff --git a/README.md b/README.md index 350ba3e..0c4232f 100644 --- a/README.md +++ b/README.md @@ -76,15 +76,17 @@ It uses the pre-trained model specified by model name to generate an image embed **Parameters:** -​ ***img***: *numpy.ndarray* +​ ***img***: *[towhee.types.Image]()* -​ The decoded image data in numpy.ndarray. +​ The decoded image data in towhee.types.Image (numpy.ndarray). -**Returns**: *numpy.ndarray* +**Returns**: -​ The image embedding extracted by model. +​ *numpy.ndarray* + +​ The image embedding extracted by model.