diff --git a/README.md b/README.md index 229279c..22e3b75 100644 --- a/README.md +++ b/README.md @@ -10,17 +10,17 @@ In computer vision (CV) directions, image transformations are usually an indispe ## Interface -`Class TransformImage(Operator)` [source](./transform_image.py) +`Class TransformImage(Operator)` [\[source\]](./transform_image.py) `__init__(self, size: int)` -**param:** +**params:** -- size(int): the image size to use. +- size(int): the size of the output image. `__call__(self, img_tensor: Union[np.ndarray, Image.Image, torch.Tensor, str])` -**param:** +**params:** Img_tensor(np.ndarray/Image.Image/torch.Tensor, str): original image data, the type can be np.ndarry, PIL.image, or str path of the image. @@ -36,11 +36,8 @@ You can get the required python package by [requirements.txt](./requirements.txt - towhee - torch - - torchvision - - numpy - - pillow ### How it works