logo

towhee
/
image-embedding-operator-template
repo-copy-icon

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
Tree: baf284c0ee
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'baf284c0ee'
${ noResults }
image-embedding-operator-te.../test_image_embedding_operat...

9 lines
338 B
Raw Normal View History

Update Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
3 years ago
from image_embedding_operator_template import ImageEmbeddingOperatorTemplate
Update the test script Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
3 years ago
def test_image_embedding_operator(model_name='resnet50', img_path='./test_data/test.jpg', dimension=2048):
Update the test script Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
3 years ago
op = ResnetImageEmbedding(model_name)
embedding = op(img_path)
# get the output shape in README
assert (dimension,)==embedding[0].shape