towhee
/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions
13 lines
382 B
13 lines
382 B
name: 'image-embedding-operator-template'
|
|
labels:
|
|
recommended_framework: 'pytorch1.2.0' #the recommend framework, such as
|
|
class: 'image-embedding'
|
|
others: 'resnet50' #your model name
|
|
operator: 'towhee/image-embedding-operator-template' #your operator repo name
|
|
init:
|
|
model_name: str
|
|
call:
|
|
input:
|
|
img_tensor: torch.Tensor
|
|
output:
|
|
feature_vector: numpy.ndarray
|