logo
Browse Source

Update YAML

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
shiyu22 3 years ago
parent
commit
bb58aec577
  1. 4
      transform_image_operator_template.yaml

4
transform_image_operator_template.yaml

@ -3,11 +3,11 @@ labels:
recommended_framework: #the recommend framework, such as 'pytorch1.2.0' recommended_framework: #the recommend framework, such as 'pytorch1.2.0'
class: image-transform class: image-transform
others: others:
operator: 'towhee/transform-image-operator-class'
operator: 'towhee/transform-image-operator-template'
init: init:
size: int size: int
call: call:
input: input:
img: Union[np.ndarray, Image.Image, torch.Tensor, str]
img: Union[Image.Image, torch.Tensor, str]
output: output:
img_transformed: torch.Tensor img_transformed: torch.Tensor
Loading…
Cancel
Save