towhee
/
vit-image-embedding
copied
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
14 lines
240 B
14 lines
240 B
3 years ago
|
name: 'vit-embedding'
|
||
|
labels:
|
||
|
recommended_framework: pytorch1.2.0
|
||
|
class: vit-embedding
|
||
|
others: vit
|
||
|
operator: 'towhee/vit-embedding'
|
||
|
init:
|
||
|
model_name: str
|
||
|
call:
|
||
|
input:
|
||
|
img_path: str
|
||
|
output:
|
||
|
feature_vector: numpy.ndarray
|