Browse Source
Update YAML
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
1 changed files with
2 additions and
2 deletions
-
transform_image_operator_template.yaml
|
|
|
@ -3,11 +3,11 @@ labels: |
|
|
|
recommended_framework: #the recommend framework, such as 'pytorch1.2.0' |
|
|
|
class: image-transform |
|
|
|
others: |
|
|
|
operator: 'towhee/transform-image-operator-class' |
|
|
|
operator: 'towhee/transform-image-operator-template' |
|
|
|
init: |
|
|
|
size: int |
|
|
|
call: |
|
|
|
input: |
|
|
|
img: Union[np.ndarray, Image.Image, torch.Tensor, str] |
|
|
|
img: Union[Image.Image, torch.Tensor, str] |
|
|
|
output: |
|
|
|
img_transformed: torch.Tensor |
